Browse Source

Fix DisableNoise node.

pull/3216/head
comfyanonymous 8 months ago
parent
commit
d8dea4cdb8
  1. 4
      comfy_extras/nodes_custom_sampler.py

4
comfy_extras/nodes_custom_sampler.py

@ -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:
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(),)

Loading…
Cancel
Save