diff --git a/comfy/model_management.py b/comfy/model_management.py index 46178e8f4..6a909fdff 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -273,9 +273,7 @@ def get_torch_device_name(device): try: torch_device_name = get_torch_device_name(get_torch_device()) - - - + if "[ZLUDA]" in torch_device_name: print(" ") print("***----------------------ZLUDA--------------------------***") @@ -288,6 +286,7 @@ try: print("***-----------------------------------------------------***") print(" :: Device:", torch_device_name) print(" ") + except: print("Could not pick default device.")