wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-09 04:52:32 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

worst PR ever, really.

Browse Source
This commit is contained in:
Christopher Anderson 2025-08-25 09:42:27 +10:00
parent 33c43b68c3
commit 184fa5921f
1 changed files with 24 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
comfy/customzluda/zluda.py
View File

@ -537,23 +537,23 @@ class DeviceProperties:
# # ------------------- Audio Ops Patch ------------------- # # ------------------- Audio Ops Patch -------------------
# if is_zluda: # if is_zluda:
# _torch_stft = torch.stft # _torch_stft = torch.stft
# _torch_istft = torch.istft # _torch_istft = torch.istft
# def z_stft(input: torch.Tensor, window: torch.Tensor, *args, **kwargs): # 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) # 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): # 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) # return _torch_istft(input=input.cpu(), window=window.cpu(), *args, **kwargs).to(input.device)
# def z_jit(f, *_, **__): # def z_jit(f, *_, **__):
# f.graph = torch._C.Graph() # f.graph = torch._C.Graph()
# return f # return f
# torch._dynamo.config.suppress_errors = True # torch._dynamo.config.suppress_errors = True
# torch.stft = z_stft # torch.stft = z_stft
# torch.istft = z_istft # torch.istft = z_istft
# torch.jit.script = z_jit # torch.jit.script = z_jit
# # ------------------- End Audio Patch ------------------- # # ------------------- End Audio Patch -------------------
# ------------------- Top-K Fallback Patch ------------------- # ------------------- Top-K Fallback Patch -------------------