diff --git a/comfy/zluda.py b/comfy/zluda.py
index 1005fcdfc..4d83a213c 100644
--- a/comfy/zluda.py
+++ b/comfy/zluda.py
@@ -9,20 +9,48 @@ for path_ in paths:
     if "rocm" not in path_.lower():
         paths_no_rocm.append(path_)
 os.environ["PATH"]=";".join(paths_no_rocm)
+# hide rocm and hip end
 
+# fix cublast errors for newer zluda versions "CUDA error: CUBLAS_STATUS_NOT_SUPPORTED when calling `cublasLtMatmulAlgoGetHeuristic" , comment it out if you have a working hipblast setup.
+os.environ['DISABLE_ADDMM_CUDA_LT'] = '1'
+        
 import torch
 
+#audio patch
+import torch._dynamo
+
+if torch.cuda.is_available() and torch.cuda.get_device_name().endswith("[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_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
+    
+    torch._dynamo.config.suppress_errors = True
+    torch.stft = z_stft
+    torch.istft = z_istft
+    torch.jit.script = z_jit
+#audio patch end
+    
+# zluda
 if torch.cuda.get_device_name().endswith("[ZLUDA]"):
     print(" ")
     print("***----------------------ZLUDA-----------------------------***")
-    print("  ::  ZLUDA detected, disabling non-supported functions.")
+    print("  ::  ZLUDA detected, disabling non-supported functions.      ")
     torch.backends.cudnn.enabled = False
-    print("  ::  CuDNN, flash_sdp, math_sdp, mem_efficient_sdp disabled) ")
+    print("  ::  CuDNN, flash_sdp, mem_efficient_sdp disabled).          ")
     torch.backends.cuda.enable_flash_sdp(False)
     torch.backends.cuda.enable_math_sdp(True)
     torch.backends.cuda.enable_mem_efficient_sdp(False)
     print("***--------------------------------------------------------***")
-    print("  ::  Device:", torch.cuda.get_device_name())
+    print("  ::  Device : ", torch.cuda.get_device_name())
     print(" ")
 else:
     print("  ::  ZLUDA isn't detected, please try patching it.")