fixed causvid scheduler

2025-07-16 18:09:06 +02:00
parent 49aaa12689
commit a356c6af4b
6 changed files with 10 additions and 26 deletions
--- a/flux/sampling.py
+++ b/flux/sampling.py
@@ -30,8 +30,8 @@ def get_noise(
        2 * math.ceil(height / 16),
        2 * math.ceil(width / 16),
        dtype=dtype,
-        generator=torch.Generator(device="cuda").manual_seed(seed),
-    ).to(device)
+        generator=torch.Generator(device=device).manual_seed(seed),
+    )


 def prepare(t5: HFEmbedder, clip: HFEmbedder, img: Tensor, prompt: str | list[str]) -> dict[str, Tensor]: