From 9aeff135b243c27690740ee79dba4273b0bab5fd Mon Sep 17 00:00:00 2001
From: patientx <patient.x@gmail.com>
Date: Mon, 2 Jun 2025 02:55:19 +0300
Subject: [PATCH] Update zluda.py

---
 comfy/customzluda/zluda.py | 32 ++++++++++++++++----------------
 1 file changed, 16 insertions(+), 16 deletions(-)

diff --git a/comfy/customzluda/zluda.py b/comfy/customzluda/zluda.py
index 9c55de3c7..3cd6ea40b 100644
--- a/comfy/customzluda/zluda.py
+++ b/comfy/customzluda/zluda.py
@@ -192,26 +192,26 @@ class DeviceProperties:
             return DeviceProperties.PROPERTIES_OVERRIDE[name]
         return getattr(self.internal, name)
 
-# ------------------- Audio Ops Patch -------------------
-if is_zluda:
-    _torch_stft = torch.stft
-    _torch_istft = torch.istft
+# # ------------------- Audio Ops Patch -------------------
+# if is_zluda:
+    # _torch_stft = torch.stft
+    # _torch_istft = torch.istft
 
-    def z_stft(input: torch.Tensor, window: torch.Tensor, *args, **kwargs):
-        return _torch_stft(input=input.cpu(), window=window.cpu(), *args, **kwargs).to(input.device)
+    # def z_stft(input: torch.Tensor, window: torch.Tensor, *args, **kwargs):
+        # return _torch_stft(input=input.cpu(), window=window.cpu(), *args, **kwargs).to(input.device)
 
-    def z_istft(input: torch.Tensor, window: torch.Tensor, *args, **kwargs):
-        return _torch_istft(input=input.cpu(), window=window.cpu(), *args, **kwargs).to(input.device)
+    # def z_istft(input: torch.Tensor, window: torch.Tensor, *args, **kwargs):
+        # return _torch_istft(input=input.cpu(), window=window.cpu(), *args, **kwargs).to(input.device)
 
-    def z_jit(f, *_, **__):
-        f.graph = torch._C.Graph()
-        return f
+    # def z_jit(f, *_, **__):
+        # f.graph = torch._C.Graph()
+        # return f
 
-    torch._dynamo.config.suppress_errors = True
-    torch.stft = z_stft
-    torch.istft = z_istft
-    torch.jit.script = z_jit
-# ------------------- End Audio Patch -------------------
+    # torch._dynamo.config.suppress_errors = True
+    # torch.stft = z_stft
+    # torch.istft = z_istft
+    # torch.jit.script = z_jit
+# # ------------------- End Audio Patch ------------------- 
 
 # ------------------- Top-K Fallback Patch -------------------
 if is_zluda: