Upload folder using huggingface_hub

core/model_manager.py

@@ -23,6 +23,7 @@ class ModelManager:
         if hasattr(self, 'initialized'):
             return
         self.loaded_models: Dict[str, Any] = {}
+        self.last_active_loras: List[Dict[str, Any]] = []
         self.initialized = True
         print("✅ ModelManager initialized.")
 
@@ -128,7 +129,7 @@ class ModelManager:
         required_set = set(required_models)
         current_set = set(self.loaded_models.keys())
 
-        loras_changed = len(active_loras) > 0 or len(current_set - required_set) > 0
+        loras_changed = active_loras != self.last_active_loras
 
         models_to_unload = current_set - required_set
         if models_to_unload or loras_changed:
@@ -136,7 +137,10 @@ class ModelManager:
                 print(f"--- [ModelManager] Models to unload: {models_to_unload} ---")
             if loras_changed and not models_to_unload:
                 models_to_unload = current_set.intersection(required_set)
-                print(f"--- [ModelManager] LoRA configuration changed. Reloading base model(s): {models_to_unload} ---")
+                if active_loras:
+                    print(f"--- [ModelManager] LoRA configuration changed. Reloading base model(s): {models_to_unload} ---")
+                else:
+                    print(f"--- [ModelManager] LoRAs removed. Reloading base model(s) to clear patches: {models_to_unload} ---")
                 
             model_management.unload_all_models()
             self.loaded_models.clear()
@@ -158,6 +162,7 @@ class ModelManager:
         else:
              print(f"--- [ModelManager] All required models are already loaded. ---")
 
+        self.last_active_loras = active_loras
         return {name: self.loaded_models[name] for name in required_models}
 
 model_manager = ModelManager()

ui/layout.py

@@ -26,6 +26,7 @@ def build_ui(event_handler_function):
             "This demo is a streamlined version of the [Comfy web UI](https://github.com/RioShiina47/comfy-webui)'s ImgGen functionality. "
             "Other versions are also available: "
             "[Qwen-Image](https://huggingface.co/spaces/RioShiina/ImageGen-Qwen-Image), "
+            "[NewBie-Image](https://huggingface.co/spaces/RioShiina/ImageGen-NewBie-Image), "
             "[Illstrious](https://huggingface.co/spaces/RioShiina/ImageGen-Illstrious), "
             "[NoobAI](https://huggingface.co/spaces/RioShiina/ImageGen-NoobAI), "
             "[Pony](https://huggingface.co/spaces/RioShiina/ImageGen-Pony1), "