Upload 5 files

app.py

@@ -134,6 +134,7 @@ flux_pipe = FluxPipeline.from_pretrained(
 ).to("cuda")
 components = flux_pipe.components
 components.pop("transformer", None)
+components.pop("scheduler", None)
 delete_model(flux_repo)
 # components = None
 
@@ -226,10 +227,7 @@ class GuiSD:
         yield f"Loading model: {model_name}"
 
         if vae_model == "BakedVAE":
-            if not os.path.exists(model_name):
-                vae_model = model_name
-            else:
-                vae_model = None
+            vae_model = model_name
         elif vae_model:
             vae_type = "SDXL" if "sdxl" in vae_model.lower() else "SD 1.5"
             if model_type != vae_type:
@@ -453,10 +451,10 @@ class GuiSD:
         self.model.stream_config(concurrency=concurrency, latent_resize_by=1, vae_decoding=False)
 
         if task != "txt2img" and not image_control:
-            raise ValueError("No control image found: To use this function, you have to upload an image in 'Image ControlNet/Inpaint/Img2img'")
+            raise ValueError("Reference image is required. Please upload one in 'Image ControlNet/Inpaint/Img2img'.")
 
-        if task == "inpaint" and not image_mask:
-            raise ValueError("No mask image found: Specify one in 'Image Mask'")
+        if task in ["inpaint", "repaint"] and not image_mask:
+            raise ValueError("Mask image not found. Upload one in 'Image Mask' to proceed.")
 
         if "https://" not in str(UPSCALER_DICT_GUI[upscaler_model_path]):
             upscaler_model = upscaler_model_path

constants.py

@@ -30,6 +30,7 @@ LOAD_DIFFUSERS_FORMAT_MODEL = [
     'shauray/FluxDev-HyperSD-merged',
     'mikeyandfriends/PixelWave_FLUX.1-dev_03',
     'terminusresearch/FluxBooru-v0.3',
+    'black-forest-labs/FLUX.1-schnell',
     'ostris/OpenFLUX.1',
     'shuttleai/shuttle-3-diffusion',
     'Laxhar/noobai-XL-1.0',
@@ -45,12 +46,15 @@ LOAD_DIFFUSERS_FORMAT_MODEL = [
     'John6666/ntr-mix-illustrious-xl-noob-xl-v777-sdxl',
     'John6666/ntr-mix-illustrious-xl-noob-xl-v777forlora-sdxl',
     'John6666/ntr-mix-illustrious-xl-noob-xl-xi-sdxl',
+    'John6666/ntr-mix-illustrious-xl-noob-xl-xii-sdxl',
+    'John6666/ntr-mix-illustrious-xl-noob-xl-xiii-sdxl',
     'John6666/mistoon-anime-v10illustrious-sdxl',
     'John6666/hassaku-xl-illustrious-v10-sdxl',
     'John6666/hassaku-xl-illustrious-v10style-sdxl',
     'John6666/haruki-mix-illustrious-v10-sdxl',
     'John6666/noobreal-v10-sdxl',
     'John6666/complicated-noobai-merge-vprediction-sdxl',
+    'Laxhar/noobai-XL-Vpred-0.9r',
     'Laxhar/noobai-XL-Vpred-0.75s',
     'Laxhar/noobai-XL-Vpred-0.75',
     'Laxhar/noobai-XL-Vpred-0.65s',
@@ -77,6 +81,7 @@ LOAD_DIFFUSERS_FORMAT_MODEL = [
     'John6666/meinaxl-v2-sdxl',
     'Eugeoter/artiwaifu-diffusion-2.0',
     'comin/IterComp',
+    'John6666/epicrealism-xl-vxiabeast-sdxl',
     'John6666/epicrealism-xl-v10kiss2-sdxl',
     'John6666/epicrealism-xl-v8kiss-sdxl',
     'misri/zavychromaxl_v80',
@@ -109,13 +114,17 @@ LOAD_DIFFUSERS_FORMAT_MODEL = [
     'John6666/cyberrealistic-pony-v63-sdxl',
     'John6666/cyberrealistic-pony-v64-sdxl',
     'John6666/cyberrealistic-pony-v65-sdxl',
+    'John6666/cyberrealistic-pony-v7-sdxl',
     'GraydientPlatformAPI/realcartoon-pony-diffusion',
     'John6666/nova-anime-xl-pony-v5-sdxl',
     'John6666/autismmix-sdxl-autismmix-pony-sdxl',
     'John6666/aimz-dream-real-pony-mix-v3-sdxl',
+    'John6666/prefectious-xl-nsfw-v10-sdxl',
+    'GraydientPlatformAPI/iniverseponyRealGuofeng49',
     'John6666/duchaiten-pony-real-v11fix-sdxl',
     'John6666/duchaiten-pony-real-v20-sdxl',
     'John6666/duchaiten-pony-xl-no-score-v70-sdxl',
+    'Spestly/OdysseyXL-3.0',
     'KBlueLeaf/Kohaku-XL-Zeta',
     'cagliostrolab/animagine-xl-3.1',
     'yodayo-ai/kivotos-xl-2.0',
@@ -130,7 +139,9 @@ LOAD_DIFFUSERS_FORMAT_MODEL = [
     'digiplay/darkphoenix3D_v1.1',
     'digiplay/BeenYouLiteL11_diffusers',
     'GraydientPlatformAPI/rev-animated2',
-    'youknownothing/cyberrealistic_v50',
+    'myxlmynx/cyberrealistic_classic40',
+    'GraydientPlatformAPI/cyberreal6',
+    'GraydientPlatformAPI/cyberreal5',
     'youknownothing/deliberate-v6',
     'GraydientPlatformAPI/deliberate-cyber3',
     'GraydientPlatformAPI/picx-real',

env.py

@@ -16,6 +16,7 @@ LOAD_DIFFUSERS_FORMAT_MODEL = [
     'kayfahaarukku/irAsu-1.0',
     'Eugeoter/artiwaifu-diffusion-2.0',
     'comin/IterComp',
+    'Emanon14/NONAMEmix_v1',
     'OnomaAIResearch/Illustrious-xl-early-release-v0',
     'Raelina/Rae-Diffusion-XL-V2',
     'Raelina/Raemu-XL-V4',

requirements.txt

@@ -1,4 +1,4 @@
-git+https://github.com/R3gm/stablepy.git@5e66972  # -b refactor_sampler_fix
+git+https://github.com/R3gm/stablepy.git@47c19f5  # -b refactor_sampler_fix
 torch==2.2.0
 numpy<2
 gdown

utils.py

@@ -62,7 +62,7 @@ class ModelInformation:
         self.download_url = json_data.get("downloadUrl", "")
         self.model_url = f"https://civitai.com/models/{self.model_id}?modelVersionId={self.model_version_id}"
         self.filename_url = next(
-            (v.get("name", "") for v in reversed(json_data.get("files", [])) if str(self.model_version_id) in v.get("downloadUrl", "")), ""
+            (v.get("name", "") for v in json_data.get("files", []) if str(self.model_version_id) in v.get("downloadUrl", "") and v.get("type", "Model") == "Model"), ""
         )
         self.filename_url = self.filename_url if self.filename_url else ""
         self.description = json_data.get("description", "")
@@ -300,7 +300,7 @@ def get_model_type(repo_id: str):
     default = "SD 1.5"
     try:
         if os.path.exists(repo_id):
-            tag, _, _ = checkpoint_model_type(repo_id)
+            tag, _, _, _ = checkpoint_model_type(repo_id)
             return DIFFUSECRAFT_CHECKPOINT_NAME[tag]
         else:
             model = api.model_info(repo_id=repo_id, timeout=5.0)