From 4d07a5f8bbd9059e08c0bad090723e016ce13977 Mon Sep 17 00:00:00 2001 From: patientx Date: Mon, 22 Dec 2025 02:35:51 +0300 Subject: [PATCH] Update model_management.py --- comfy/model_management.py | 1 + 1 file changed, 1 insertion(+) diff --git a/comfy/model_management.py b/comfy/model_management.py index 1889ab0ac..45922faa3 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -27,6 +27,7 @@ import platform import weakref import gc import os +import comfy.zluda class VRAMState(Enum): DISABLED = 0 #No vram present: no need to move models to vram