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

Merge pull request #465 from DefTruth/main

Browse Source 
[Bug] fix parallel rand device & set dynamic cfg
This commit is contained in:
Yuxuan.Zhang 2024-11-08 13:32:57 +08:00 committed by GitHub
commit 075fad4dae
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tools/parallel_inference/parallel_inference_xdit.py
View File

@ -78,8 +78,9 @@ def main():
num_frames=input_config.num_frames, num_frames=input_config.num_frames,
prompt=input_config.prompt, prompt=input_config.prompt,
num_inference_steps=input_config.num_inference_steps, num_inference_steps=input_config.num_inference_steps,
generator=torch.Generator(device="cuda").manual_seed(input_config.seed), generator=torch.Generator().manual_seed(input_config.seed),
guidance_scale=6, guidance_scale=6,
use_dynamic_cfg=True,
).frames[0] ).frames[0]
end_time = time.time() end_time = time.time()