Views
No views yet
tencent/HunyuanVideo.1import torch
2from diffusers import HunyuanVideoPipeline, HunyuanVideoTransformer3DModel
3from diffusers.utils import export_to_video
4
5model_id = "hunyuanvideo-community/HunyuanVideo"
6transformer = HunyuanVideoTransformer3DModel.from_pretrained(
7 model_id, subfolder="transformer", torch_dtype=torch.bfloat16
8)
9pipe = HunyuanVideoPipeline.from_pretrained(model_id, transformer=transformer, torch_dtype=torch.float16)
10
11# Enable memory savings
12pipe.vae.enable_tiling()
13pipe.enable_model_cpu_offload()
14
15output = pipe(
16 prompt="A cat walks on the grass, realistic",
17 height=320,
18 width=512,
19 num_frames=61,
20 num_inference_steps=30,
21).frames[0]
22export_to_video(output, "output.mp4", fps=15)