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

Update model_management.py

Browse Source
This commit is contained in:
patientx 2024-12-03 01:35:06 +03:00 committed by GitHub
parent 9dea868c65
commit 1d7cbcdcb2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/model_management.py
View File

@ -274,8 +274,6 @@ def get_torch_device_name(device):
try: try:
torch_device_name = get_torch_device_name(get_torch_device()) torch_device_name = get_torch_device_name(get_torch_device())
if "[ZLUDA]" in torch_device_name: if "[ZLUDA]" in torch_device_name:
print(" ") print(" ")
print("***----------------------ZLUDA--------------------------***") print("***----------------------ZLUDA--------------------------***")
@ -288,6 +286,7 @@ try:
print("***-----------------------------------------------------***") print("***-----------------------------------------------------***")
print(" :: Device:", torch_device_name) print(" :: Device:", torch_device_name)
print(" ") print(" ")
except: except:
print("Could not pick default device.") print("Could not pick default device.")