Skip to content

Commit 075fad4

Browse files
Merge pull request #465 from DefTruth/main
[Bug] fix parallel rand device & set dynamic cfg
2 parents abe334f + 1e6d1bb commit 075fad4

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)