From 90ba86a95fd78e86cdf3ada7df2983340d92096e Mon Sep 17 00:00:00 2001 From: Christopher Anderson Date: Sat, 8 Nov 2025 16:12:45 +1100 Subject: [PATCH] Add env TORCH_AMD_CUDNN_ENABLED (import os) --- comfy/model_management.py | 1 + 1 file changed, 1 insertion(+) diff --git a/comfy/model_management.py b/comfy/model_management.py index 949d21dbd..8c2aa5179 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -26,6 +26,7 @@ import importlib import platform import weakref import gc +import os class VRAMState(Enum): DISABLED = 0 #No vram present: no need to move models to vram