Fix noise with ancestral samplers when inferencing on cpu. (#11528)

This commit is contained in:
comfyanonymous 2025-12-26 19:03:01 -08:00 committed by GitHub
parent 16fb6849d2
commit 1e4e342f54
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -74,6 +74,9 @@ def get_ancestral_step(sigma_from, sigma_to, eta=1.):
def default_noise_sampler(x, seed=None): def default_noise_sampler(x, seed=None):
if seed is not None: if seed is not None:
if x.device == torch.device("cpu"):
seed += 1
generator = torch.Generator(device=x.device) generator = torch.Generator(device=x.device)
generator.manual_seed(seed) generator.manual_seed(seed)
else: else: