workaround for unload_all_models removing LoRA

2025-01-25 20:21:59 +02:00
parent 30188f670a
commit 1c966f0ee7
1 changed files with 4 additions and 0 deletions
@@ -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