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.")
 
@@ -145,7 +146,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 = self.last_active_loras != active_loras
 
         models_to_unload = current_set - required_set
         if models_to_unload or loras_changed:
@@ -171,7 +172,10 @@ class ModelManager:
                     loaded_model_data = self._load_model_combo(display_name, active_loras, progress)
                     self.loaded_models[display_name] = loaded_model_data
                 except Exception as e:
+                    self.last_active_loras = []
                     raise gr.Error(f"Failed to load model combo or apply LoRA for '{display_name}'. Reason: {e}")
+            
+            self.last_active_loras = active_loras
         else:
              print(f"--- [ModelManager] All required models are already loaded. ---")
 

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: "
             "[Z-Image](https://huggingface.co/spaces/RioShiina/ImageGen-Z-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), "