From 6f01b244a29e7b15b4a33168ceb4d603a6244c86 Mon Sep 17 00:00:00 2001
From: rattus <46076784+rattus128@users.noreply.github.com>
Date: Wed, 10 Jun 2026 03:57:04 +1000
Subject: [PATCH] mm: dont reset cast buffers in cleanup_models_gc() (#14372)

cleanup_models_gc can be called once per load_models_gpu via
free_memory, which in turn can de-activate an active model via
this reset_cast_buffers.

cleanup_models_gc() could also come via obscure garbage collector
paths so limit reset_cast_buffers to the post-node callsite instead.
---
 comfy/model_management.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/comfy/model_management.py b/comfy/model_management.py
index 8e786c0a5..9dc0a4e13 100644
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -958,8 +958,6 @@ def loaded_models(only_currently_used=False):
 def cleanup_models_gc():
     do_gc = False
 
-    reset_cast_buffers()
-
     for i in range(len(current_loaded_models)):
         cur = current_loaded_models[i]
         if cur.is_dead():