Skip to content

Commit 11c15d8

Browse files
Fix fp16 intermediates giving different results. (Comfy-Org#13100)
1 parent b5d32e6 commit 11c15d8

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

comfy/sample.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,12 +8,12 @@
88

99
def prepare_noise_inner(latent_image, generator, noise_inds=None):
1010
if noise_inds is None:
11-
return torch.randn(latent_image.size(), dtype=latent_image.dtype, layout=latent_image.layout, generator=generator, device="cpu")
11+
return torch.randn(latent_image.size(), dtype=torch.float32, layout=latent_image.layout, generator=generator, device="cpu").to(dtype=latent_image.dtype)
1212

1313
unique_inds, inverse = np.unique(noise_inds, return_inverse=True)
1414
noises = []
1515
for i in range(unique_inds[-1]+1):
16-
noise = torch.randn([1] + list(latent_image.size())[1:], dtype=latent_image.dtype, layout=latent_image.layout, generator=generator, device="cpu")
16+
noise = torch.randn([1] + list(latent_image.size())[1:], dtype=torch.float32, layout=latent_image.layout, generator=generator, device="cpu").to(dtype=latent_image.dtype)
1717
if i in unique_inds:
1818
noises.append(noise)
1919
noises = [noises[i] for i in inverse]

0 commit comments

Comments
 (0)