Skip to content

Commit 1e6d1bb

Browse files
committed
fix parallel rand device
1 parent 4aebdb4 commit 1e6d1bb

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tools/parallel_inference/parallel_inference_xdit.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,8 +78,9 @@ def main():
7878
num_frames=input_config.num_frames,
7979
prompt=input_config.prompt,
8080
num_inference_steps=input_config.num_inference_steps,
81-
generator=torch.Generator(device="cuda").manual_seed(input_config.seed),
81+
generator=torch.Generator().manual_seed(input_config.seed),
8282
guidance_scale=6,
83+
use_dynamic_cfg=True,
8384
).frames[0]
8485

8586
end_time = time.time()

0 commit comments

Comments
 (0)