Upload 7 files

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

@@ -47,6 +47,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',
     'Laxhar/noobai-XL-1.0',
     'Raelina/Rae-Diffusion-XL-V2',

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
 accelerate
 diffusers
 invisible_watermark

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)