Fix noise with ancestral samplers when inferencing on cpu.

2026-07-21 23:41:28 +08:00 · 2025-12-26 21:54:08 -05:00
parent 16fb6849d2
commit 2e005aef1d
1 changed files with 3 additions and 0 deletions
@@ -74,6 +74,9 @@ def get_ancestral_step(sigma_from, sigma_to, eta=1.):

 def default_noise_sampler(x, seed=None):
    if seed is not None:
+        if x.device == torch.device("cpu"):
+            seed += 1
+
        generator = torch.Generator(device=x.device)
        generator.manual_seed(seed)
    else: