From 894c727ce26454b7e0bb0eca22698048860705f7 Mon Sep 17 00:00:00 2001 From: patientx Date: Thu, 5 Sep 2024 00:05:54 +0300 Subject: [PATCH] Update model_management.py --- comfy/model_management.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/comfy/model_management.py b/comfy/model_management.py index 01f4ac4d7..237fa9999 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -275,19 +275,17 @@ try: if "[ZLUDA]" in torch_device_name: + print(" ") 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.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(" :: Device:", torch_device_name) + print(" ") except: print("Could not pick default device.")