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

Update model_management.py

Browse Source
This commit is contained in:
patientx 2024-09-05 00:05:54 +03:00 committed by GitHub
parent 1b7b7ad6f1
commit 894c727ce2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
comfy/model_management.py
View File

@ -275,19 +275,17 @@ try:
if "[ZLUDA]" in torch_device_name: if "[ZLUDA]" in torch_device_name:
print(" ")
print("***----------------------ZLUDA--------------------------***") print("***----------------------ZLUDA--------------------------***")
print(" :: ZLUDA detected, disabling non-supported functions.") print(" :: ZLUDA detected, disabling non-supported functions.")
torch.backends.cudnn.enabled = False
if torch.backends.cudnn.enabled: print(" :: (cuDNN, flash_sdp, mem_efficient_sdp disabled) ")
torch.backends.cudnn.enabled = False
print(" :: cuDNN , flash_sdp , mem_efficient_sdp disabled")
torch.backends.cuda.enable_flash_sdp(False) torch.backends.cuda.enable_flash_sdp(False)
torch.backends.cuda.enable_math_sdp(True) torch.backends.cuda.enable_math_sdp(True)
torch.backends.cuda.enable_mem_efficient_sdp(False) torch.backends.cuda.enable_mem_efficient_sdp(False)
print("***-----------------------------------------------------***") print("***-----------------------------------------------------***")
print(" :: Device:", torch_device_name)
print("Device:", torch_device_name) print(" ")
except: except:
print("Could not pick default device.") print("Could not pick default device.")