theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-20 23:58:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

hunyuanvideo_vae_decoder

Browse Source
This commit is contained in:
mi804
2024-12-18 11:14:57 +08:00
parent 5d1005a7c8
commit 263166768e
4 changed files with 41 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/video_synthesis/hunyuanvideo.py
View File

@@ -1,4 +1,4 @@
from diffsynth import ModelManager, HunyuanVideoPipeline, download_models
from diffsynth import ModelManager, HunyuanVideoPipeline, download_models, save_video
import torch
@@ -8,9 +8,11 @@ download_models(["HunyuanVideo"])
# Load models
model_manager = ModelManager(torch_dtype=torch.float16, device="cuda")
model_manager.load_models([
"models/HunyuanVideo/vae/pytorch_model.pt",
"t2i_models/HunyuanVideo/text_encoder/model.safetensors",
"t2i_models/HunyuanVideo/text_encoder_2",
])
pipe = HunyuanVideoPipeline.from_model_manager(model_manager)
prompt = 'A cat walks on the grass, realistic style.'
pipe(prompt)
frames = pipe(prompt)
save_video(frames, 'test_video.mp4', fps=8, quality=5)