Unofficial community fork for Diffusers-format weights on tencent/HunyuanVideo
.
Using Diffusers
HunyuanVideo can be used directly from Diffusers. Install the latest version of Diffusers.
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 for more information.
- Downloads last month
- 128
Model tree for hunyuanvideo-community/HunyuanVideo
Base model
tencent/HunyuanVideo