|
--- |
|
base_model: |
|
- tencent/HunyuanVideo |
|
library_name: diffusers |
|
--- |
|
|
|
Unofficial community fork for Diffusers-format weights on [`tencent/HunyuanVideo`](https://huggingface.co/tencent/HunyuanVideo). |
|
|
|
### Using Diffusers |
|
|
|
HunyuanVideo can be used directly from Diffusers. Install the latest version of Diffusers. |
|
|
|
```python |
|
import torch |
|
from diffusers import HunyuanVideoPipeline, HunyuanVideoTransformer3DModel |
|
from diffusers.utils import export_to_video |
|
|
|
model_id = "hunyuanvideo-community/HunyuanVideo" |
|
transformer = HunyuanVideoTransformer3DModel.from_pretrained( |
|
model_id, subfolder="transformer", torch_dtype=torch.bfloat16 |
|
) |
|
pipe = HunyuanVideoPipeline.from_pretrained(model_id, transformer=transformer, torch_dtype=torch.float16) |
|
|
|
# Enable memory savings |
|
pipe.vae.enable_tiling() |
|
pipe.enable_model_cpu_offload() |
|
|
|
output = pipe( |
|
prompt="A cat walks on the grass, realistic", |
|
height=320, |
|
width=512, |
|
num_frames=61, |
|
num_inference_steps=30, |
|
).frames[0] |
|
export_to_video(output, "output.mp4", fps=15) |
|
``` |
|
|
|
Refer to the [documentation](https://huggingface.co/docs/diffusers/main/en/api/pipelines/hunyuan_video) for more information. |
|
|
|
|
