Commit d8dea4cd authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix DisableNoise node.

parent a7dd82e6
......@@ -320,7 +320,7 @@ class Noise_EmptyNoise:
def generate_noise(self, input_latent):
latent_image = input_latent["samples"]
return torch.zeros(shape, dtype=latent_image.dtype, layout=latent_image.layout, device="cpu")
return torch.zeros(latent_image.shape, dtype=latent_image.dtype, layout=latent_image.layout, device="cpu")
class Noise_RandomNoise:
......@@ -477,7 +477,7 @@ class DisableNoise:
FUNCTION = "get_noise"
CATEGORY = "sampling/custom_sampling/noise"
def get_noise(self, noise_seed):
def get_noise(self):
return (Noise_EmptyNoise(),)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment