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.")
 
@@ -89,15 +90,16 @@ 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:
+        
+        must_reload = bool(models_to_unload) or loras_changed
+
+        if must_reload:
             if models_to_unload:
                 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} ---")
+                print(f"--- [ModelManager] LoRA configuration changed. Reloading base model(s): {current_set.intersection(required_set)} ---")
                 
             model_management.unload_all_models()
             self.loaded_models.clear()
@@ -105,7 +107,7 @@ class ModelManager:
             torch.cuda.empty_cache()
             print("--- [ModelManager] All models unloaded to free RAM. ---")
         
-        models_to_load = required_set if (models_to_unload or loras_changed) else required_set - current_set
+        models_to_load = required_set if must_reload else (required_set - current_set)
         
         if models_to_load:
             print(f"--- [ModelManager] Models to load: {models_to_load} ---")
@@ -136,7 +138,8 @@ class ModelManager:
                     raise gr.Error(f"Failed to load model or apply LoRA '{display_name}'. Reason: {e}")
         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

@@ -27,6 +27,7 @@ def build_ui(event_handler_function):
             "Other versions are also available: "
             "[Z-Image](https://huggingface.co/spaces/RioShiina/ImageGen-Z-Image), "
             "[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), "