From 0683dc912c2cf22cf2e43f996a43e444416dbefd Mon Sep 17 00:00:00 2001 From: LeagueRaINi Date: Mon, 26 Feb 2024 02:32:21 +0100 Subject: [PATCH] feat: pytorch attention does not seem to crash the driver anymore --- comfy/model_management.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/comfy/model_management.py b/comfy/model_management.py index 33313ac68..5e656b827 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -267,10 +267,6 @@ try: torch.backends.cuda.enable_math_sdp(True) torch.backends.cuda.enable_mem_efficient_sdp(False) - if ENABLE_PYTORCH_ATTENTION: - print("Disabling pytorch cross attention because ZLUDA does currently not support it.") - ENABLE_PYTORCH_ATTENTION = False - print("Device:", torch_device_name) except: print("Could not pick default device.")