wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-23 21:00:16 +08:00
Code Issues Actions 13 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

48
comfy/customzluda/zluda.py
View File

@ -71,7 +71,7 @@ def detect_amd_gpu_architecture():
try: try:
import subprocess import subprocess
result = subprocess.run(['wmic', 'path', 'win32_VideoController', 'get', 'name'], result = subprocess.run(['wmic', 'path', 'win32_VideoController', 'get', 'name'],
capture_output=True, text=True, timeout=10) capture_output=True, text=True, timeout=10)
if result.returncode == 0: if result.returncode == 0:
for line in result.stdout.split('\n'): for line in result.stdout.split('\n'):
line = line.strip() line = line.strip()
@ -303,10 +303,10 @@ def handle_pydantic_packages(required_packages):
try: try:
print(" :: Installing compatible pydantic packages...") print(" :: Installing compatible pydantic packages...")
combined_args = [sys.executable, '-m', 'pip', 'install', combined_args = [sys.executable, '-m', 'pip', 'install',
'pydantic~=2.0', 'pydantic~=2.0',
'pydantic-settings~=2.0', 'pydantic-settings~=2.0',
'--quiet', '--quiet',
'--disable-pip-version-check'] '--disable-pip-version-check']
subprocess.check_call(combined_args) subprocess.check_call(combined_args)
@ -344,10 +344,10 @@ def install_package(package_name, version_spec, upgrade=False):
try: try:
print(f" :: Retrying {package_name} installation without version constraint...") print(f" :: Retrying {package_name} installation without version constraint...")
fallback_args = [sys.executable, '-m', 'pip', 'install', fallback_args = [sys.executable, '-m', 'pip', 'install',
package_name, package_name,
'--upgrade', '--upgrade',
'--quiet', '--quiet',
'--disable-pip-version-check'] '--disable-pip-version-check']
subprocess.check_call(fallback_args) subprocess.check_call(fallback_args)
print(f" :: {package_name} installed successfully without version constraint") print(f" :: {package_name} installed successfully without version constraint")
except subprocess.CalledProcessError as e2: except subprocess.CalledProcessError as e2:
@ -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 -------------------
@ -637,8 +637,8 @@ def do_hijack():
def amd_flash_wrapper(query, key, value, attn_mask=None, dropout_p=0.0, is_causal=False, scale=None): def amd_flash_wrapper(query, key, value, attn_mask=None, dropout_p=0.0, is_causal=False, scale=None):
try: try:
if (query.shape[-1] <= 128 and if (query.shape[-1] <= 128 and
attn_mask is None and # fix flash-attention error : "Flash attention error: Boolean value of Tensor with more than one value is ambiguous" attn_mask is None and # fix flash-attention error : "Flash attention error: Boolean value of Tensor with more than one value is ambiguous"
query.dtype != torch.float32): query.dtype != torch.float32):
if scale is None: if scale is None:
scale = query.shape[-1] ** -0.5 scale = query.shape[-1] ** -0.5
return interface_fa.fwd( return interface_fa.fwd(