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

Update stepvideo examples

Browse Source
This commit is contained in:
Zhongjie Duan
2025-02-18 18:00:08 +08:00
committed by GitHub
parent 9cff769fbd
commit 2899283c01
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/stepvideo/stepvideo_text_to_video_low_vram.py
View File

@@ -28,7 +28,7 @@ model_manager.load_models(
"models/stepfun-ai/stepvideo-t2v/transformer/diffusion_pytorch_model-00006-of-00006.safetensors", "models/stepfun-ai/stepvideo-t2v/transformer/diffusion_pytorch_model-00006-of-00006.safetensors",
] ]
], ],
torch_dtype=torch.float8_e4m3fn, device="cpu" torch_dtype=torch.bfloat16, device="cpu" # You can set torch_dtype=torch.bfloat16 to reduce RAM (not VRAM) usage.
) )
model_manager.load_models( model_manager.load_models(
["models/stepfun-ai/stepvideo-t2v/vae/vae_v2.safetensors"], ["models/stepfun-ai/stepvideo-t2v/vae/vae_v2.safetensors"],