wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-28 23:30:16 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

UPDATED ZLUDA PATCHING METHOD

Browse Source
This commit is contained in:
patientx 2025-01-14 19:57:22 +03:00 committed by GitHub
parent c3fc894ce2
commit c4861c74d4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
comfy/model_management.py
View File

@ -25,6 +25,7 @@ import sys
import platform import platform
import weakref import weakref
import gc import gc
import zluda
class VRAMState(Enum): class VRAMState(Enum):
DISABLED = 0 #No vram present: no need to move models to vram DISABLED = 0 #No vram present: no need to move models to vram
@ -301,23 +302,9 @@ def get_torch_device_name(device):
return "CUDA {}: {}".format(device, torch.cuda.get_device_name(device)) return "CUDA {}: {}".format(device, torch.cuda.get_device_name(device))
try: try:
torch_device_name = get_torch_device_name(get_torch_device()) logging.info("Device: {}".format(get_torch_device_name(get_torch_device())))
if "[ZLUDA]" in torch_device_name:
print(" ")
print("***----------------------ZLUDA--------------------------***")
print(" :: ZLUDA detected, disabling non-supported functions.")
torch.backends.cudnn.enabled = False
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_device_name)
print(" ")
except: except:
print("Could not pick default device.") logging.warning("Could not pick default device.")
current_loaded_models = [] current_loaded_models = []