wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-22 12:20:16 +08:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

Update model_management.py

Browse Source
This commit is contained in:
patientx 2024-08-30 20:10:47 +03:00 committed by GitHub
parent c371e1da3a
commit 51af2440ef
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 25 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
comfy/model_management.py
View File

@ -44,13 +44,13 @@ cpu_state = CPUState.GPU
total_vram = 0 total_vram = 0
xpu_available = False #We don't have xpu with zluda , so disabling all together. xpu_available = False
torch_version = "" torch_version = ""
try: try:
torch_version = torch.version.__version__ torch_version = torch.version.__version__
xpu_available = (int(torch_version[0]) < 2 or (int(torch_version[0]) == 2 and int(torch_version[2]) <= 4)) and torch.xpu.is_available() xpu_available = (int(torch_version[0]) < 2 or (int(torch_version[0]) == 2 and int(torch_version[2]) <= 4)) and torch.xpu.is_available()
except: except:
pass pass
lowvram_available = True lowvram_available = True
if args.deterministic: if args.deterministic:
@ -270,24 +270,26 @@ 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("***----------------------ZLUDA--------------------------***")
print(" :: ZLUDA detected, disabling non-supported functions.")
if torch.backends.cudnn.enabled:
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)
except: except:
print("Could not pick default device.") logging.warning("Could not pick default device.")
current_loaded_models = [] current_loaded_models = []