From d8dea4cdb88cae92ac4b4a4d79bbfd60b7f8923d Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Fri, 5 Apr 2024 21:36:23 -0400
Subject: [PATCH] Fix DisableNoise node.

---
 comfy_extras/nodes_custom_sampler.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/comfy_extras/nodes_custom_sampler.py b/comfy_extras/nodes_custom_sampler.py
index a5d791d9c..56cb505e6 100644
--- a/comfy_extras/nodes_custom_sampler.py
+++ b/comfy_extras/nodes_custom_sampler.py
@@ -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(),)