Upload 6 files

README.md

@@ -11,6 +11,8 @@ header: mini
 license: mit
 duplicated_from: r3gm/DiffuseCraft
 short_description: Stunning images using stable diffusion.
+preload_from_hub:
+  - madebyollin/sdxl-vae-fp16-fix config.json,diffusion_pytorch_model.safetensors
 hf_oauth: true
 ---
 

constants.py

@@ -7,10 +7,10 @@ from stablepy import (
 )
 
 # - **Download Models**
-DOWNLOAD_MODEL = "https://civitai.com/api/download/models/574369, https://huggingface.co/TechnoByte/MilkyWonderland/resolve/main/milkyWonderland_v40.safetensors"
+DOWNLOAD_MODEL = "https://huggingface.co/TechnoByte/MilkyWonderland/resolve/main/milkyWonderland_v40.safetensors"
 
 # - **Download VAEs**
-DOWNLOAD_VAE = "https://huggingface.co/nubby/blessed-sdxl-vae-fp16-fix/resolve/main/sdxl_vae-fp16fix-c-1.1-b-0.5.safetensors?download=true, https://huggingface.co/nubby/blessed-sdxl-vae-fp16-fix/resolve/main/sdxl_vae-fp16fix-blessed.safetensors?download=true, https://huggingface.co/digiplay/VAE/resolve/main/vividReal_v20.safetensors?download=true, https://huggingface.co/fp16-guy/anything_kl-f8-anime2_vae-ft-mse-840000-ema-pruned_blessed_clearvae_fp16_cleaned/resolve/main/vae-ft-mse-840000-ema-pruned_fp16.safetensors?download=true"
+DOWNLOAD_VAE = "https://huggingface.co/fp16-guy/anything_kl-f8-anime2_vae-ft-mse-840000-ema-pruned_blessed_clearvae_fp16_cleaned/resolve/main/vae-ft-mse-840000-ema-pruned_fp16.safetensors?download=true"
 
 # - **Download LoRAs**
 DOWNLOAD_LORA = "https://huggingface.co/Leopain/color/resolve/main/Coloring_book_-_LineArt.safetensors, https://civitai.com/api/download/models/135867, https://huggingface.co/Linaqruf/anime-detailer-xl-lora/resolve/main/anime-detailer-xl.safetensors?download=true, https://huggingface.co/Linaqruf/style-enhancer-xl-lora/resolve/main/style-enhancer-xl.safetensors?download=true, https://huggingface.co/ByteDance/Hyper-SD/resolve/main/Hyper-SD15-8steps-CFG-lora.safetensors?download=true, https://huggingface.co/ByteDance/Hyper-SD/resolve/main/Hyper-SDXL-8steps-CFG-lora.safetensors?download=true"
@@ -24,7 +24,9 @@ LOAD_DIFFUSERS_FORMAT_MODEL = [
     'John6666/lyh-anime-flux-v2a1-fp8-flux',
     'John6666/carnival-unchained-v10-fp8-flux',
     'Freepik/flux.1-lite-8B-alpha',
-    'ostris/OpenFLUX.1',
+    'shauray/FluxDev-HyperSD-merged',
+    'mikeyandfriends/PixelWave_FLUX.1-dev_03',
+    'terminusresearch/FluxBooru-v0.3',
     'John6666/noobai-xl-nai-xl-epsilonpred10version-sdxl',
     'John6666/noobai-xl-nai-xl-epsilonpred075version-sdxl',
     'John6666/noobai-xl-nai-xl-epsilonpred05version-sdxl',
@@ -38,6 +40,7 @@ LOAD_DIFFUSERS_FORMAT_MODEL = [
     'John6666/illustriousxl-mmmix-v50-sdxl',
     'John6666/illustrious-pencil-xl-v200-sdxl',
     'John6666/obsession-illustriousxl-v21-sdxl',
+    'John6666/obsession-illustriousxl-v30-sdxl',
     'eienmojiki/Anything-XL',
     'eienmojiki/Starry-XL-v5.2',
     'John6666/meinaxl-v2-sdxl',
@@ -86,6 +89,8 @@ LOAD_DIFFUSERS_FORMAT_MODEL = [
     'yodayo-ai/kivotos-xl-2.0',
     'yodayo-ai/holodayo-xl-2.1',
     'yodayo-ai/clandestine-xl-1.0',
+    'ostris/OpenFLUX.1',
+    'shuttleai/shuttle-3-diffusion',
     'digiplay/majicMIX_sombre_v2',
     'digiplay/majicMIX_realistic_v6',
     'digiplay/majicMIX_realistic_v7',
@@ -120,8 +125,8 @@ DIFFUSERS_FORMAT_LORAS = [
 
 DOWNLOAD_EMBEDS = [
     'https://huggingface.co/datasets/Nerfgun3/bad_prompt/blob/main/bad_prompt_version2.pt',
-    'https://huggingface.co/embed/negative/resolve/main/EasyNegativeV2.safetensors',
-    'https://huggingface.co/embed/negative/resolve/main/bad-hands-5.pt',
+    # 'https://huggingface.co/embed/negative/resolve/main/EasyNegativeV2.safetensors',
+    # 'https://huggingface.co/embed/negative/resolve/main/bad-hands-5.pt',
     ]
 
 CIVITAI_API_KEY = os.environ.get("CIVITAI_API_KEY")
@@ -291,6 +296,13 @@ MODEL_TYPE_CLASS = {
     "diffusers:FluxPipeline": "FLUX",
 }
 
+DIFFUSECRAFT_CHECKPOINT_NAME = {
+    "sd1.5": "SD 1.5",
+    "sdxl": "SDXL",
+    "flux-dev": "FLUX",
+    "flux-schnell": "FLUX",
+}
+
 POST_PROCESSING_SAMPLER = ["Use same sampler"] + [
     name_s for name_s in scheduler_names if "Auto-Loader" not in name_s
 ]

env.py

@@ -1,103 +1,12 @@
 import os
-
+from constants import LOAD_DIFFUSERS_FORMAT_MODEL as LOAD_DIFFUSERS_FORMAT_MODEL_DC
 CIVITAI_API_KEY = os.environ.get("CIVITAI_API_KEY")
 HF_TOKEN = os.environ.get("HF_TOKEN")
 HF_READ_TOKEN = os.environ.get('HF_READ_TOKEN') # only use for private repo
 
 # - **List Models**
 LOAD_DIFFUSERS_FORMAT_MODEL = [
-    'stabilityai/stable-diffusion-xl-base-1.0',
-    'black-forest-labs/FLUX.1-dev',
-    'John6666/blue-pencil-flux1-v021-fp8-flux',
-    'John6666/wai-ani-flux-v10forfp8-fp8-flux',
-    'John6666/xe-anime-flux-v04-fp8-flux',
-    'John6666/lyh-anime-flux-v2a1-fp8-flux',
-    'John6666/carnival-unchained-v10-fp8-flux',
-    'Freepik/flux.1-lite-8B-alpha',
-    'ostris/OpenFLUX.1',
-    'John6666/noobai-xl-nai-xl-epsilonpred10version-sdxl',
-    'John6666/noobai-xl-nai-xl-epsilonpred075version-sdxl',
-    'John6666/noobai-xl-nai-xl-epsilonpred05version-sdxl',
-    'John6666/noobai-cyberfix-v10-sdxl',
-    'John6666/noobaiiter-xl-vpred-v075-sdxl',
-    'John6666/complicated-noobai-merge-vprediction-sdxl',
-    'John6666/noobai-fusion2-vpred-itercomp-v1-sdxl',
-    'John6666/noobai-xl-nai-xl-vpredtestversion-sdxl',
-    'John6666/chadmix-noobai075-illustrious01-v10-sdxl',
-    'OnomaAIResearch/Illustrious-xl-early-release-v0',
-    'John6666/obsession-illustriousxl-v21-sdxl',
-    'eienmojiki/Anything-XL',
-    'eienmojiki/Starry-XL-v5.2',
-    'John6666/meinaxl-v2-sdxl',
-    'John6666/epicrealism-xl-v10kiss2-sdxl',
-    'John6666/epicrealism-xl-v8kiss-sdxl',
-    'misri/zavychromaxl_v80',
-    'SG161222/RealVisXL_V4.0',
-    'SG161222/RealVisXL_V5.0',
-    'misri/newrealityxlAllInOne_Newreality40',
-    'gsdf/CounterfeitXL',
-    'John6666/silvermoon-mix-01xl-v11-sdxl',
-    'WhiteAiZ/autismmixSDXL_autismmixConfetti_diffusers',
-    'kitty7779/ponyDiffusionV6XL',
-    'GraydientPlatformAPI/aniverse-pony',
-    'John6666/ras-real-anime-screencap-v1-sdxl',
-    'John6666/duchaiten-pony-xl-no-score-v60-sdxl',
-    'John6666/mistoon-anime-ponyalpha-sdxl',
-    'John6666/3x3x3mixxl-v2-sdxl',
-    'John6666/3x3x3mixxl-3dv01-sdxl',
-    'John6666/ebara-mfcg-pony-mix-v12-sdxl',
-    'John6666/t-ponynai3-v51-sdxl',
-    'John6666/t-ponynai3-v65-sdxl',
-    'John6666/prefect-pony-xl-v3-sdxl',
-    'John6666/mala-anime-mix-nsfw-pony-xl-v5-sdxl',
-    'John6666/wai-real-mix-v11-sdxl',
-    'John6666/wai-c-v6-sdxl',
-    'John6666/iniverse-mix-xl-sfwnsfw-pony-guofeng-v43-sdxl',
-    'John6666/sifw-annihilation-xl-v2-sdxl',
-    'John6666/photo-realistic-pony-v5-sdxl',
-    'John6666/pony-realism-v21main-sdxl',
-    'John6666/pony-realism-v22main-sdxl',
-    'John6666/cyberrealistic-pony-v63-sdxl',
-    'John6666/cyberrealistic-pony-v64-sdxl',
-    'John6666/cyberrealistic-pony-v65-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/duchaiten-pony-real-v11fix-sdxl',
-    'John6666/duchaiten-pony-real-v20-sdxl',
-    'John6666/duchaiten-pony-xl-no-score-v70-sdxl',
-    'Eugeoter/artiwaifu-diffusion-2.0',
-    'comin/IterComp',
-    'KBlueLeaf/Kohaku-XL-Zeta',
-    'cagliostrolab/animagine-xl-3.1',
-    'yodayo-ai/kivotos-xl-2.0',
-    'yodayo-ai/holodayo-xl-2.1',
-    'yodayo-ai/clandestine-xl-1.0',
-    'digiplay/majicMIX_sombre_v2',
-    'digiplay/majicMIX_realistic_v6',
-    'digiplay/majicMIX_realistic_v7',
-    'digiplay/DreamShaper_8',
-    'digiplay/BeautifulArt_v1',
-    'digiplay/DarkSushi2.5D_v1',
-    'digiplay/darkphoenix3D_v1.1',
-    'digiplay/BeenYouLiteL11_diffusers',
-    'Yntec/RevAnimatedV2Rebirth',
-    'youknownothing/cyberrealistic_v50',
-    'youknownothing/deliberate-v6',
-    'GraydientPlatformAPI/deliberate-cyber3',
-    'GraydientPlatformAPI/picx-real',
-    'GraydientPlatformAPI/perfectworld6',
-    'emilianJR/epiCRealism',
-    'votepurchase/counterfeitV30_v30',
-    'votepurchase/ChilloutMix',
-    'Meina/MeinaMix_V11',
-    'Meina/MeinaUnreal_V5',
-    'Meina/MeinaPastel_V7',
-    'GraydientPlatformAPI/realcartoon3d-17',
-    'GraydientPlatformAPI/realcartoon-pixar11',
-    'GraydientPlatformAPI/realcartoon-real17',
-    'nitrosocke/Ghibli-Diffusion',
+    'Laxhar/noobai-XL-1.0',
     'KBlueLeaf/Kohaku-XL-Epsilon-rev2',
     'KBlueLeaf/Kohaku-XL-Epsilon-rev3',
     'KBlueLeaf/Kohaku-XL-Zeta',
@@ -105,7 +14,7 @@ LOAD_DIFFUSERS_FORMAT_MODEL = [
     'kayfahaarukku/irAsu-1.0',
     'Eugeoter/artiwaifu-diffusion-2.0',
     'comin/IterComp',
-    "OnomaAIResearch/Illustrious-xl-early-release-v0",
+    'OnomaAIResearch/Illustrious-xl-early-release-v0',
     'Raelina/Rae-Diffusion-XL-V2',
     'Raelina/Raemu-XL-V4',
     'Raelina/Raehoshi-illust-XL',
@@ -118,17 +27,18 @@ LOAD_DIFFUSERS_FORMAT_MODEL = [
     "multimodalart/FLUX.1-dev2pro-full",
     "Raelina/Raemu-Flux",
 ]
+LOAD_DIFFUSERS_FORMAT_MODEL = LOAD_DIFFUSERS_FORMAT_MODEL_DC + LOAD_DIFFUSERS_FORMAT_MODEL
 
 DIFFUSERS_FORMAT_LORAS = [
     "nerijs/animation2k-flux",
     "XLabs-AI/flux-RealismLora",
+    "Shakker-Labs/FLUX.1-dev-LoRA-Logo-Design",
 ]
 
 # List all Models for specified user
 HF_MODEL_USER_LIKES = ["votepurchase"] # sorted by number of likes
 HF_MODEL_USER_EX = ["John6666"] # sorted by a special rule
 
-
 # - **Download Models**
 DOWNLOAD_MODEL_LIST = [
 ]

requirements.txt

@@ -1,4 +1,4 @@
-git+https://github.com/R3gm/stablepy.git@ed51089 # -b refactor_sampler_fix
+git+https://github.com/R3gm/stablepy.git@f686946  # -b refactor_sampler_fix
 torch==2.2.0
 gdown
 opencv-python

utils.py

@@ -7,11 +7,14 @@ from constants import (
     HF_TOKEN,
     MODEL_TYPE_CLASS,
     DIRECTORY_LORAS,
+    DIFFUSECRAFT_CHECKPOINT_NAME,
 )
 from huggingface_hub import HfApi
+from huggingface_hub import snapshot_download
 from diffusers import DiffusionPipeline
 from huggingface_hub import model_info as model_info_data
 from diffusers.pipelines.pipeline_loading_utils import variant_compatible_siblings
+from stablepy.diffusers_vanilla.utils import checkpoint_model_type
 from pathlib import PosixPath
 from unidecode import unidecode
 import urllib.parse
@@ -283,10 +286,15 @@ def get_model_type(repo_id: str):
     api = HfApi(token=os.environ.get("HF_TOKEN"))  # if use private or gated model
     default = "SD 1.5"
     try:
-        model = api.model_info(repo_id=repo_id, timeout=5.0)
-        tags = model.tags
-        for tag in tags:
-            if tag in MODEL_TYPE_CLASS.keys(): return MODEL_TYPE_CLASS.get(tag, default)
+        if os.path.exists(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)
+            tags = model.tags
+            for tag in tags:
+                if tag in MODEL_TYPE_CLASS.keys(): return MODEL_TYPE_CLASS.get(tag, default)
+
     except Exception:
         return default
     return default
@@ -371,17 +379,23 @@ def download_diffuser_repo(repo_name: str, model_type: str, revision: str = "mai
         if len(variant_filenames):
             variant = "fp16"
 
-    cached_folder = DiffusionPipeline.download(
-        pretrained_model_name=repo_name,
-        force_download=False,
-        token=token,
-        revision=revision,
-        # mirror="https://hf-mirror.com",
-        variant=variant,
-        use_safetensors=True,
-        trust_remote_code=False,
-        timeout=5.0,
-    )
+    if model_type == "FLUX":
+        cached_folder = snapshot_download(
+            repo_id=repo_name,
+            allow_patterns="transformer/*"
+        )
+    else:
+        cached_folder = DiffusionPipeline.download(
+            pretrained_model_name=repo_name,
+            force_download=False,
+            token=token,
+            revision=revision,
+            # mirror="https://hf-mirror.com",
+            variant=variant,
+            use_safetensors=True,
+            trust_remote_code=False,
+            timeout=5.0,
+        )
 
     if isinstance(cached_folder, PosixPath):
         cached_folder = cached_folder.as_posix()