JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-04-06 03:57:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update cli_demo.py

Browse Source
This commit is contained in:
zR 2024-08-15 14:26:37 +08:00
parent f666fd4e95
commit 1af3e5bfc4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference/cli_demo.py
View File

@ -90,7 +90,7 @@ def generate_video(
prompt=prompt,
num_videos_per_prompt=num_videos_per_prompt, # Number of videos to generate per prompt
num_inference_steps=num_inference_steps, # Number of inference steps
num_frames=49,
num_frames=48, # Number of frames to generate，changed to 49 for diffusers version `0.31.0` and after.
guidance_scale=guidance_scale, # Guidance scale for classifier-free guidance
generator=torch.Generator().manual_seed(42), # Set the seed for reproducibility
).frames[0]