JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-11-15 22:42:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix parallel rand device

Browse Source
This commit is contained in:
DefTruth 2024-11-07 02:40:13 +00:00
parent 4aebdb4b66
commit 1e6d1bbb82
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()