From 1c966f0ee712a877afddd99c86eeb6a1be56c514 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sat, 25 Jan 2025 20:21:59 +0200
Subject: [PATCH] workaround for unload_all_models removing LoRA

---
 nodes.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/nodes.py b/nodes.py
index 3d02b96..016ca37 100644
--- a/nodes.py
+++ b/nodes.py
@@ -507,6 +507,10 @@ class HyVideoModelLoader:
         patcher.model["auto_cpu_offload"] = auto_cpu_offload
         patcher.model["scheduler_config"] = scheduler_config
 
+        for model in mm.current_loaded_models:
+            if model._model() == patcher:
+                mm.current_loaded_models.remove(model)            
+
         return (patcher,)
 
 #region load VAE