Skip to content

GitLab

Commit d8dea4cd authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix DisableNoise node.

parent a7dd82e6
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
comfy_extras/nodes_custom_sampler.py
View file @ d8dea4cd
...@@ -320,7 +320,7 @@ class Noise_EmptyNoise: ...@@ -320,7 +320,7 @@ class Noise_EmptyNoise:
def generate_noise(self, input_latent): def generate_noise(self, input_latent):
latent_image = input_latent["samples"] 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: class Noise_RandomNoise:
...@@ -477,7 +477,7 @@ class DisableNoise: ...@@ -477,7 +477,7 @@ class DisableNoise:
FUNCTION = "get_noise" FUNCTION = "get_noise"
CATEGORY = "sampling/custom_sampling/noise" CATEGORY = "sampling/custom_sampling/noise"
def get_noise(self, noise_seed): def get_noise(self):
return (Noise_EmptyNoise(),) 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