298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-03-15 14:09:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix DisableNoise node.

Browse Source
This commit is contained in:
comfyanonymous 2024-04-05 21:36:23 -04:00
parent a7dd82e668
commit d8dea4cdb8
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_extras/nodes_custom_sampler.py
View File

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