Fix mistake. (#10484)

2025-10-25 20:07:29 -07:00
parent 098a352f13
commit f6bbc1ac84
1 changed files with 1 additions and 1 deletions
--- a/comfy/sample.py
+++ b/comfy/sample.py
@@ -17,7 +17,7 @@ def prepare_noise_inner(latent_image, generator, noise_inds=None):
        if i in unique_inds:
            noises.append(noise)
    noises = [noises[i] for i in inverse]
-    noises = torch.cat(noises, axis=0)
+    return torch.cat(noises, axis=0)

 def prepare_noise(latent_image, seed, noise_inds=None):
    """