Upload 9 files

constants.py

@@ -17,22 +17,34 @@ DOWNLOAD_LORA = "https://huggingface.co/Leopain/color/resolve/main/Coloring_book
 
 LOAD_DIFFUSERS_FORMAT_MODEL = [
     'stabilityai/stable-diffusion-xl-base-1.0',
+    'Laxhar/noobai-XL-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',
+    'John6666/iniverse-mix-xl-sfwnsfw-fluxdfp16nsfwv11-fp8-flux',
     'Freepik/flux.1-lite-8B-alpha',
     'shauray/FluxDev-HyperSD-merged',
     'mikeyandfriends/PixelWave_FLUX.1-dev_03',
     'terminusresearch/FluxBooru-v0.3',
+    'ostris/OpenFLUX.1',
+    'shuttleai/shuttle-3-diffusion',
     'John6666/noobai-xl-nai-xl-epsilonpred10version-sdxl',
+    'Laxhar/noobai-XL-0.77',
     'John6666/noobai-xl-nai-xl-epsilonpred075version-sdxl',
+    'Laxhar/noobai-XL-0.6',
     'John6666/noobai-xl-nai-xl-epsilonpred05version-sdxl',
     'John6666/noobai-cyberfix-v10-sdxl',
     'John6666/noobaiiter-xl-vpred-v075-sdxl',
+    'John6666/ntr-mix-illustrious-xl-noob-xl-v40-sdxl',
+    'John6666/ntr-mix-illustrious-xl-noob-xl-ntrmix35-sdxl',
+    'John6666/haruki-mix-illustrious-v10-sdxl',
+    'John6666/noobreal-v10-sdxl',
     'John6666/complicated-noobai-merge-vprediction-sdxl',
+    'Laxhar/noobai-XL-Vpred-0.6',
+    'John6666/noobai-xl-nai-xl-vpred05version-sdxl',
     'John6666/noobai-fusion2-vpred-itercomp-v1-sdxl',
     'John6666/noobai-xl-nai-xl-vpredtestversion-sdxl',
     'John6666/chadmix-noobai075-illustrious01-v10-sdxl',
@@ -41,9 +53,15 @@ LOAD_DIFFUSERS_FORMAT_MODEL = [
     'John6666/illustrious-pencil-xl-v200-sdxl',
     'John6666/obsession-illustriousxl-v21-sdxl',
     'John6666/obsession-illustriousxl-v30-sdxl',
+    'John6666/wai-nsfw-illustrious-v70-sdxl',
+    'John6666/illustrious-pony-mix-v3-sdxl',
+    'John6666/nova-anime-xl-illustriousv10-sdxl',
+    'John6666/silvermoon-mix03-illustrious-v10-sdxl',
     'eienmojiki/Anything-XL',
     'eienmojiki/Starry-XL-v5.2',
     'John6666/meinaxl-v2-sdxl',
+    'Eugeoter/artiwaifu-diffusion-2.0',
+    'comin/IterComp',
     'John6666/epicrealism-xl-v10kiss2-sdxl',
     'John6666/epicrealism-xl-v8kiss-sdxl',
     'misri/zavychromaxl_v80',
@@ -51,21 +69,22 @@ LOAD_DIFFUSERS_FORMAT_MODEL = [
     '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/prefect-pony-xl-v4-sdxl',
     'John6666/mala-anime-mix-nsfw-pony-xl-v5-sdxl',
+    'John6666/wai-ani-nsfw-ponyxl-v10-sdxl',
+    'John6666/wai-ani-nsfw-ponyxl-v9-sdxl',
     'John6666/wai-real-mix-v11-sdxl',
+    'John6666/babes-by-stable-yogi-ponyv3-sdxl',
     'John6666/wai-c-v6-sdxl',
     'John6666/iniverse-mix-xl-sfwnsfw-pony-guofeng-v43-sdxl',
     'John6666/sifw-annihilation-xl-v2-sdxl',
@@ -82,15 +101,11 @@ LOAD_DIFFUSERS_FORMAT_MODEL = [
     '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',
-    'ostris/OpenFLUX.1',
-    'shuttleai/shuttle-3-diffusion',
     'digiplay/majicMIX_sombre_v2',
     'digiplay/majicMIX_realistic_v6',
     'digiplay/majicMIX_realistic_v7',
@@ -137,6 +152,9 @@ DIRECTORY_LORAS = 'loras'
 DIRECTORY_VAES = 'vaes'
 DIRECTORY_EMBEDS = 'embedings'
 
+CACHE_HF = "/home/user/.cache/huggingface/hub/"
+STORAGE_ROOT = "/home/user/"
+
 PREPROCESSOR_CONTROLNET = {
   "openpose": [
     "Openpose",
@@ -424,7 +442,7 @@ EXAMPLES_GUI = [
         0.35,  # strength
         1.0,  # cn scale
         0.05,  # cn start
-        0.75,  # cn end
+        0.8,  # cn end
         "Classic",
         None,
         35,

dc.py

@@ -36,6 +36,8 @@ from utils import (
     extract_exif_data,
     create_mask_now,
     download_diffuser_repo,
+    get_used_storage_gb,
+    delete_model,
     progress_step_bar,
     html_template_message,
     escape_html,
@@ -109,8 +111,8 @@ for url_embed in DOWNLOAD_EMBEDS:
 
 # Build list models
 embed_list = get_model_list(DIRECTORY_EMBEDS)
-model_list = get_model_list(DIRECTORY_MODELS)
-model_list = load_diffusers_format_model + model_list
+single_file_model_list = get_model_list(DIRECTORY_MODELS)
+model_list = list_uniq(get_model_id_list() + LOAD_DIFFUSERS_FORMAT_MODEL + single_file_model_list)
 
 ## BEGIN MOD
 lora_model_list = get_lora_model_list()
@@ -135,6 +137,21 @@ class GuiSD:
         self.status_loading = False
         self.sleep_loading = 4
         self.last_load = datetime.now()
+        self.inventory = []
+
+    def update_storage_models(self, storage_floor_gb=42, required_inventory_for_purge=3):
+        while get_used_storage_gb() > storage_floor_gb:
+            if len(self.inventory) < required_inventory_for_purge:
+                break
+            removal_candidate = self.inventory.pop(0)
+            delete_model(removal_candidate)
+
+    def update_inventory(self, model_name):
+        if model_name not in single_file_model_list:
+            self.inventory = [
+                m for m in self.inventory if m != model_name
+            ] + [model_name]
+        print(self.inventory)
 
     def infer_short(self, model, pipe_params, progress=gr.Progress(track_tqdm=True)):
         #progress(0, desc="Start inference...")
@@ -148,6 +165,11 @@ class GuiSD:
         return img
 
     def load_new_model(self, model_name, vae_model, task, progress=gr.Progress(track_tqdm=True)):
+
+        self.update_storage_models()
+
+        # download link model > model_name
+
         vae_model = vae_model if vae_model != "None" else None
         model_type = get_model_type(model_name)
         dtype_model = torch.bfloat16 if model_type == "FLUX" else torch.float16
@@ -160,6 +182,8 @@ class GuiSD:
                 token=True,
             )
 
+        self.update_inventory(model_name)
+
         for i in range(68):
             if not self.status_loading:
                 self.status_loading = True

llmdolphin.py

@@ -79,6 +79,52 @@ llm_models = {
     "ChatWaifu_22B_v2.0_preview.Q4_K_S.gguf": ["mradermacher/ChatWaifu_22B_v2.0_preview-GGUF", MessagesFormatterType.MISTRAL],
     "ChatWaifu_v1.4.Q5_K_M.gguf": ["mradermacher/ChatWaifu_v1.4-GGUF", MessagesFormatterType.MISTRAL],
     "ChatWaifu_v1.3.1.Q4_K_M.gguf": ["mradermacher/ChatWaifu_v1.3.1-GGUF", MessagesFormatterType.MISTRAL],
+    "SeQwence-14B-v5.Q4_K_S.gguf": ["mradermacher/SeQwence-14B-v5-GGUF", MessagesFormatterType.OPEN_CHAT],
+    "L3.1-8B-Dark-Planet-Slush.i1-Q4_K_M.gguf": ["mradermacher/L3.1-8B-Dark-Planet-Slush-i1-GGUF", MessagesFormatterType.LLAMA_3],
+    "QwenSlerp12-7B.Q5_K_M.gguf": ["mradermacher/QwenSlerp12-7B-GGUF", MessagesFormatterType.OPEN_CHAT],
+    "qwen-carpmuscle-v0.4.Q4_K_M.gguf": ["mradermacher/qwen-carpmuscle-v0.4-GGUF", MessagesFormatterType.OPEN_CHAT],
+    "L3.1-8B-Slush-v1.1.i1-Q4_K_M.gguf": ["mradermacher/L3.1-8B-Slush-v1.1-i1-GGUF", MessagesFormatterType.LLAMA_3],
+    "Llama-3.2-8B-Instruct.Q5_K_M.gguf": ["mradermacher/Llama-3.2-8B-Instruct-GGUF", MessagesFormatterType.LLAMA_3],
+    "Homer-v0.3-Qwen2.5-7B.Q5_K_M.gguf": ["mradermacher/Homer-v0.3-Qwen2.5-7B-GGUF", MessagesFormatterType.OPEN_CHAT],
+    "Norns-Qwen2.5-7B-v0.2.i1-Q5_K_M.gguf": ["mradermacher/Norns-Qwen2.5-7B-v0.2-i1-GGUF", MessagesFormatterType.OPEN_CHAT],
+    "LLAMA-3_8B_Unaligned_BETA.i1-Q5_K_M.gguf": ["mradermacher/LLAMA-3_8B_Unaligned_BETA-i1-GGUF", MessagesFormatterType.LLAMA_3],
+    "SELM-Llama-3-8B-Instruct-iter-3.Q5_K_M.gguf": ["mradermacher/SELM-Llama-3-8B-Instruct-iter-3-GGUF", MessagesFormatterType.LLAMA_3],
+    "ArliAI-RPMax-v1.3-merge-8B.i1-Q4_K_M.gguf": ["mradermacher/ArliAI-RPMax-v1.3-merge-8B-i1-GGUF", MessagesFormatterType.LLAMA_3],
+    "gemma-2-9b-it-DS-V2.Q4_K_M.gguf": ["mradermacher/gemma-2-9b-it-DS-V2-GGUF", MessagesFormatterType.ALPACA],
+    "Stheno-Hercules-3.1-8B.i1-Q5_K_M.gguf": ["mradermacher/Stheno-Hercules-3.1-8B-i1-GGUF", MessagesFormatterType.LLAMA_3],
+    "EVA-Norns-Qwen2.5-v0.1.i1-Q5_K_M.gguf": ["mradermacher/EVA-Norns-Qwen2.5-v0.1-i1-GGUF", MessagesFormatterType.OPEN_CHAT],
+    "Infinirc-ArliAI-RPMax-v1.3-merge-8B.i1-Q5_K_M.gguf": ["mradermacher/Infinirc-ArliAI-RPMax-v1.3-merge-8B-i1-GGUF", MessagesFormatterType.LLAMA_3],
+    "Verdandi-Qwen2.5-7B.Q5_K_M.gguf": ["mradermacher/Verdandi-Qwen2.5-7B-GGUF", MessagesFormatterType.OPEN_CHAT],
+    "Qwen2-Wukong-7B.Q5_K_M.gguf": ["mradermacher/Qwen2-Wukong-7B-GGUF", MessagesFormatterType.OPEN_CHAT],
+    "Lelantos-7B.Q5_K_M.gguf": ["mradermacher/Lelantos-7B-GGUF", MessagesFormatterType.MISTRAL],
+    "Kunoichi-7B.Q5_K_M.gguf": ["mradermacher/Kunoichi-7B-GGUF", MessagesFormatterType.MISTRAL],
+    "llama-3-cat-8b-instruct-v1.Q5_K_M.gguf": ["mradermacher/llama-3-cat-8b-instruct-v1-GGUF", MessagesFormatterType.LLAMA_3],
+    "DPOpenHermes-7B.i1-Q5_K_M.gguf": ["mradermacher/DPOpenHermes-7B-i1-GGUF", MessagesFormatterType.MISTRAL],
+    "NeuralDaredevil-8B-abliterated.Q5_K_M.gguf": ["mradermacher/NeuralDaredevil-8B-abliterated-GGUF", MessagesFormatterType.LLAMA_3],
+    "AstroSage-8B.Q5_K_M.gguf": ["mradermacher/AstroSage-8B-GGUF", MessagesFormatterType.LLAMA_3],
+    "Llama-3-8B-Theresa.Q5_K_M.gguf": ["mradermacher/Llama-3-8B-Theresa-GGUF", MessagesFormatterType.LLAMA_3],
+    "Sonya-7B.Q5_K_M.gguf": ["mradermacher/Sonya-7B-GGUF", MessagesFormatterType.MISTRAL],
+    "Hathor_Tahsin-L3-8B-v0.9.Q4_K_S.gguf": ["mradermacher/Hathor_Tahsin-L3-8B-v0.9-GGUF", MessagesFormatterType.LLAMA_3],
+    "Qwen-Qwen2.5-7B-Instruct-llamafied.Q5_K_M.gguf": ["mradermacher/Qwen-Qwen2.5-7B-Instruct-llamafied-GGUF", MessagesFormatterType.OPEN_CHAT],
+    "Ice0.34n-14.11-RP.i1-Q4_K_S.gguf": ["mradermacher/Ice0.34n-14.11-RP-i1-GGUF", MessagesFormatterType.MISTRAL],
+    "Qwen2.5-14B-Wernicke-SFT.Q4_K_S.gguf": ["mradermacher/Qwen2.5-14B-Wernicke-SFT-GGUF", MessagesFormatterType.OPEN_CHAT],
+    "Qwen2.5-7B-nerd-uncensored-v1.6.i1-Q4_K_S.gguf": ["mradermacher/Qwen2.5-7B-nerd-uncensored-v1.6-i1-GGUF", MessagesFormatterType.OPEN_CHAT],
+    "ArliAI-RPMax-v1.3-merge-8B.Q4_K_S.gguf": ["mradermacher/ArliAI-RPMax-v1.3-merge-8B-GGUF", MessagesFormatterType.LLAMA_3],
+    "MN-Violet-Lotus-12B.Q4_K_M.gguf": ["mradermacher/MN-Violet-Lotus-12B-GGUF", MessagesFormatterType.MISTRAL],
+    "Qwen2.5-7B-nerd-uncensored-v1.8.Q5_K_M.gguf": ["mradermacher/Qwen2.5-7B-nerd-uncensored-v1.8-GGUF", MessagesFormatterType.OPEN_CHAT],
+    "ArliAI-RPMax-v1.3-merge-llama3-8B.i1-Q4_K_M.gguf": ["mradermacher/ArliAI-RPMax-v1.3-merge-llama3-8B-i1-GGUF", MessagesFormatterType.LLAMA_3],
+    "Llama-3.1-SuperNova-8B-Lite_TIES_with_Base.i1-Q5_K_M.gguf": ["mradermacher/Llama-3.1-SuperNova-8B-Lite_TIES_with_Base-i1-GGUF", MessagesFormatterType.LLAMA_3],
+    "maverick-llama3-8B.i1-Q4_K_M.gguf": ["mradermacher/maverick-llama3-8B-i1-GGUF", MessagesFormatterType.LLAMA_3],
+    "JSL-MedLlama-3-8B-v2.0.i1-Q5_K_M.gguf": ["mradermacher/JSL-MedLlama-3-8B-v2.0-i1-GGUF", MessagesFormatterType.LLAMA_3],
+    "hiwaifu-12b.i1-Q4_K_M.gguf": ["mradermacher/hiwaifu-12b-i1-GGUF", MessagesFormatterType.MISTRAL],
+    "samantha-1.1-westlake-7b-laser.i1-Q5_K_M.gguf": ["mradermacher/samantha-1.1-westlake-7b-laser-i1-GGUF", MessagesFormatterType.MISTRAL],
+    "Wukong-0.1-Mistral-7B-v0.2.i1-Q5_K_M.gguf": ["mradermacher/Wukong-0.1-Mistral-7B-v0.2-i1-GGUF", MessagesFormatterType.MISTRAL],
+    "InfinityRP-v1-7B.i1-Q5_K_M.gguf": ["mradermacher/InfinityRP-v1-7B-i1-GGUF", MessagesFormatterType.MISTRAL],
+    "L3-Luna-8B.i1-Q5_K_M.gguf": ["mradermacher/L3-Luna-8B-i1-GGUF", MessagesFormatterType.LLAMA_3],
+    "Awanllm-Llama-3-8B-Instruct-ORPO-v0.1.i1-Q5_K_M.gguf": ["mradermacher/Awanllm-Llama-3-8B-Instruct-ORPO-v0.1-i1-GGUF", MessagesFormatterType.LLAMA_3],
+    "dolphin-2.9.1-llama-3-8b.i1-Q5_K_M.gguf": ["mradermacher/dolphin-2.9.1-llama-3-8b-i1-GGUF", MessagesFormatterType.LLAMA_3],
+    "EVA-Tissint-14B.i1-Q4_K_M.gguf": ["mradermacher/EVA-Tissint-14B-i1-GGUF", MessagesFormatterType.OPEN_CHAT],
+    "EVA-Tissint-v1.1-14B.Q4_K_M.gguf": ["mradermacher/EVA-Tissint-v1.1-14B-GGUF", MessagesFormatterType.OPEN_CHAT],
     "magnum-12b-v2.5-kto.i1-Q4_K_M.gguf": ["mradermacher/magnum-12b-v2.5-kto-i1-GGUF", MessagesFormatterType.CHATML],
     "L3.1-8B-Slush.i1-Q5_K_M.gguf": ["mradermacher/L3.1-8B-Slush-i1-GGUF", MessagesFormatterType.LLAMA_3],
     "QandoraExp-7B-Persona.i1-Q5_K_M.gguf": ["mradermacher/QandoraExp-7B-Persona-i1-GGUF", MessagesFormatterType.OPEN_CHAT],

lora_dict.json

@@ -2519,6 +2519,13 @@
     "https://civitai.com/models/504990",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/708e949d-7dfd-473b-952a-e09317b679ca/width=450/15176927.jpeg"
   ],
+  "Valkyria_Tentacles_Pony_V2": [
+    "colorful tentacles, tentacle pit, Tentacle Restraints, tentacle floors, tentacle walls, Big tentacles, large tentacles, huge tentacles, giant tentacles, huge penis, giant penis, large insertion, Double insertion, triple insertion, inflation, lactation, stomach bulge, torn clothes, shiny, shiny skin, slime , wet skin, sweat, oral, fellatio, irrumatio,  magical girl,  wide spread legs,  / bukkake, facial, excessive cum, cum drip, cum over, cum in pussy, vomiting cum, cum through, projectile,",
+    "Pony",
+    "Hentai Tentacles Hard Core (Valkyria Style) (PONY XL) | \u89e6\u624b\u51cc\u8fb1 (\u308f\u308b\u304d\u3085\u30fc\u308c \u30b9\u30bf\u30a4\u30eb)",
+    "https://civitai.com/models/915381",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/f6ce1d3d-02f9-48ec-922c-3099838cf836/width=450/38247950.jpeg"
+  ],
   "VibingDance_(frieren_vibing)_LoRA_PDXL_Anzhc": [
     "vibingdance, animation sheet, chibi, dancing / simple background, white background",
     "Pony",
@@ -2645,6 +2652,13 @@
     "https://civitai.com/models/308865",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/0fd71659-09aa-4179-bfd2-879b623403d2/width=450/22504558.jpeg"
   ],
+  "[hikari]jellyfish_haircut_pony_v1": [
+    "jfcut, long hair",
+    "Pony",
+    "\u6c34\u6bcd\u5934\u53d1\u578b / \u30af\u30e9\u30b2\u30ab\u30c3\u30c8 / Jellyfish haircut Concept Pony LoRA[HiKaRi]",
+    "https://civitai.com/models/955239",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/33e99e1e-5aa6-481a-989a-8e74b4d1c793/width=450/40538903.jpeg"
+  ],
   "[hikari]missionary_holding_wrist_pony_v1": [
     "missionary_holding_wrist",
     "Pony",
@@ -2827,6 +2841,13 @@
     "https://civitai.com/models/614638",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/0c6ed1ad-da10-4c9a-90ba-a1d12a8ea46c/width=450/22239895.jpeg"
   ],
+  "_sdxl-style-illusioneyle": [
+    "illusioneyle",
+    "Illustrious",
+    "for Illustrious, for \"Ilustrious\", stability of illustration patterns (Illustrious\u7528 \u7d75\u67c4\u5b89\u5b9a\u6027\u5411\u4e0a\u306e\u305f\u3081\u306b) LoRA",
+    "https://civitai.com/models/944662",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/ec2e9a82-7bb7-4363-954c-2d3cc7a0bd08/width=450/39901570.jpeg"
+  ],
   "_sdxl-transchool-pony": [
     "transchool",
     "Pony",
@@ -3674,6 +3695,13 @@
     "https://civitai.com/models/812637",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/25e5c86b-a6d1-4129-99be-d37ebcd43e4b/width=450/32617666.jpeg"
   ],
+  "chikubi_karikari_v3_mt-enhanced": [
+    "chikubi_karikari, from front, nipple stimulation, motion lines, sound effects, ${your desired breasts size} / chikubi_karikari, from side, nipple stimulation, motion lines, sound effects, ${your desired breasts size} / chikubi_karikari, from below, nipple stimulation, motion lines, sound effects, ${your desired breasts size}",
+    "Pony",
+    "(CONCEPT) Nipple-Scratching / \u4e73\u9996\u30ab\u30ea\u30ab\u30ea\u30aa\u30ca\u30cb\u30fc",
+    "https://civitai.com/models/812637",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/fb20c5fe-88e0-48b7-a656-b64691ecbb2f/width=450/40471676.jpeg"
+  ],
   "china_dress_Pony_V1_0": [
     "jyojifuku, china dress, skirt",
     "Pony",
@@ -4171,6 +4199,13 @@
     "https://civitai.com/models/603258",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/a9862d57-c319-439a-a392-8b16d4adfc42/width=450/21484810.jpeg"
   ],
+  "donbiki_Illust_v1": [
+    "donbiki",
+    "Illustrious",
+    "[Illust&XL&Pony] disgust / \u30c9\u30f3\u5f15\u304d / \u990a\u8c5a\u5834\u306e\u8c5a\u3092\u898b\u308b\u3088\u3046\u306a\u76ee",
+    "https://civitai.com/models/759232",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/8c40c192-c478-40b6-bcc4-a6628d3055a1/width=450/39890508.jpeg"
+  ],
   "donbiki_Pony_v1": [
     "donbiki",
     "Pony",
@@ -4227,6 +4262,13 @@
     "https://civitai.com/models/393273",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/d61e0783-9d00-43a0-9b10-03f43ba40a51/width=450/9556301.jpeg"
   ],
+  "doyagao_xl_v10": [
+    "",
+    "Pony",
+    "[SDXL] Doyagao (V-shaped eyebrows) helper / V\u5b57\u5f62\u7709(\u30c9\u30e4\u9854\u4ed6)",
+    "https://civitai.com/models/953978",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/6d353dec-111a-439d-a978-e05ee7442593/width=450/40460311.jpeg"
+  ],
   "dress_tug": [
     "dress_tug",
     "Pony",
@@ -4934,6 +4976,41 @@
     "https://civitai.com/models/498731",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/06a0dc3a-42ba-42b5-9ea9-d6b6faa3543b/width=450/14812284.jpeg"
   ],
+  "hotarueye_xl_hosome1_v20": [
+    "",
+    "Pony",
+    "[SDXL] Semi-closed eyes / \u7d30\u76ee",
+    "https://civitai.com/models/941927",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/7b19c76a-4472-4cb8-9554-9bf031be1ce4/width=450/39739501.jpeg"
+  ],
+  "hotarueye_xl_hosome2_v20": [
+    "",
+    "Pony",
+    "[SDXL] Semi-closed eyes / \u7d30\u76ee",
+    "https://civitai.com/models/941927",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/fe620aa2-9e45-47ef-92a5-1c5c60da3be8/width=450/39739679.jpeg"
+  ],
+  "hotarueye_xl_normaljitome1_v10": [
+    "",
+    "Pony",
+    "[SDXL] Jitome(scornful eyes) / \u30b8\u30c8\u76ee",
+    "https://civitai.com/models/941966",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/684286bd-4e91-499d-9261-fade57500bb5/width=450/39741453.jpeg"
+  ],
+  "hotarueye_xl_sleepy_v10": [
+    "",
+    "Pony",
+    "[SDXL] Sleepy eyes / \u7720\u305d\u3046\u306a\u76ee",
+    "https://civitai.com/models/926566",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/5d7feca6-501b-41a6-9eca-54790f4909ef/width=450/38851516.jpeg"
+  ],
+  "hotarueye_xl_smirking_v10": [
+    "",
+    "Pony",
+    "[SDXL] Smirking eyes / \u306b\u3084\u306b\u3084\u3057\u305f\u76ee",
+    "https://civitai.com/models/926552",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/7a1bde6a-82f5-4fed-b9f4-6364de49b6d7/width=450/38850481.jpeg"
+  ],
   "hotarueye_xl_surprised_v20": [
     "",
     "Pony",
@@ -5095,6 +5172,13 @@
     "https://civitai.com/models/549854",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/69569569-e7b2-454a-81d9-4a820d628bd6/width=450/17985023.jpeg"
   ],
+  "inv_v_eyebrows_xl_v10": [
+    "",
+    "Pony",
+    "[SDXL] Inverted V-shaped eyebrows / \u30cf\u306e\u5b57\u578b\u306e\u7709",
+    "https://civitai.com/models/953992",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/11e36bc0-ef0d-415f-8a42-eb7b9d6bf9be/width=450/40461165.jpeg"
+  ],
   "japanese_girl_v1_1": [
     "jpn-girl",
     "SDXL 1.0",
@@ -5102,6 +5186,13 @@
     "https://civitai.com/models/121579",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/4d702d98-996e-478c-9e1d-0115852a3457/width=450/1899358.jpeg"
   ],
+  "jellyfish-outfit-01": [
+    "jellyfish outfit",
+    "Pony",
+    "Jellyfish Outfit",
+    "https://civitai.com/models/955236",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/c2444e84-3321-44e0-b430-2c4da65f6302/width=450/40537387.jpeg"
+  ],
   "jihanki_PONY_V1": [
     "vending machine, outdoors, scenery, coca-cola, road, street, utility pole, power lines",
     "Pony",
@@ -5354,6 +5445,13 @@
     "https://civitai.com/models/655354",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/82734312-66f2-4d1f-a7f1-6c022fb4da78/width=450/24747865.jpeg"
   ],
+  "kitsune-dance-ponyxl-lora-nochekaiser": [
+    "<lora:kitsune-dance-ponyxl-lora-nochekaiser:1>, kitsune dance, cheerleader, layered skirt, paw pose, fox ears, white belt, fox tail, baseball stadium, blue shirt, blue skirt, crop top, sleeveless shirt, midriff",
+    "Pony",
+    "Kitsune Dance (\u304d\u3064\u306d\u30c0\u30f3\u30b9) - Clothing",
+    "https://civitai.com/models/943759",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/6badb7f0-08e8-44e9-a6c7-62590eb9c511/width=450/39866024.jpeg"
+  ],
   "kk_sexybody_02": [
     "kk_sexybody_02, 1girl, solo, breasts, wet, nipples, cum on body, excessive pubic hair, nude",
     "Pony",
@@ -5781,6 +5879,13 @@
     "https://civitai.com/models/315944",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/7d391817-79ba-4afa-bd48-54ba3fce0847/width=450/6928651.jpeg"
   ],
+  "mesugaki_Illust_v1": [
+    "mesugaki",
+    "Illustrious",
+    "[Illust&XL&Pony] mesugaki / \u30e1\u30b9\u30ac\u30ad",
+    "https://civitai.com/models/455382",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/8e6a35ca-ae5b-478f-b46d-9b11fc44d77b/width=450/39100267.jpeg"
+  ],
   "mesugaki_Pony_v1": [
     "mesugaki",
     "Pony",
@@ -5991,6 +6096,13 @@
     "https://civitai.com/models/467118",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/11eced9f-d779-4138-a940-ad81a0b0454c/width=450/17635537.jpeg"
   ],
+  "noobaiXL_EasyNegative_LoRA-3_0_last": [
+    "",
+    "Illustrious",
+    "noobaiXL_EasyNegative",
+    "https://civitai.com/models/923983",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/b97ba109-85de-4cce-9bfe-1764c4eb1ef0/width=450/38759818.jpeg"
+  ],
   "nosepony1": [
     "",
     "Pony",

requirements.txt

@@ -4,8 +4,9 @@ diffusers
 invisible_watermark
 transformers
 xformers
-git+https://github.com/R3gm/stablepy.git@f686946  # -b refactor_sampler_fix
+git+https://github.com/R3gm/stablepy.git@8edabb0  # -b refactor_sampler_fix
 torch==2.2.0
+numpy<2
 gdown
 opencv-python
 huggingface_hub

utils.py

@@ -7,7 +7,10 @@ from constants import (
     HF_TOKEN,
     MODEL_TYPE_CLASS,
     DIRECTORY_LORAS,
+    DIRECTORY_MODELS,
     DIFFUSECRAFT_CHECKPOINT_NAME,
+    CACHE_HF,
+    STORAGE_ROOT,
 )
 from huggingface_hub import HfApi
 from huggingface_hub import snapshot_download
@@ -22,6 +25,8 @@ import copy
 import requests
 from requests.adapters import HTTPAdapter
 from urllib3.util import Retry
+import shutil
+import subprocess
 
 USER_AGENT = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:127.0) Gecko/20100101 Firefox/127.0'
 
@@ -110,7 +115,11 @@ def download_things(directory, url, hf_token="", civitai_api_key="", romanize=Fa
             print("\033[91mYou need an API key to download Civitai models.\033[0m")
 
         model_profile = retrieve_model_info(url)
-        if model_profile.download_url and model_profile.filename_url:
+        if (
+            model_profile is not None
+            and model_profile.download_url
+            and model_profile.filename_url
+        ):
             url = model_profile.download_url
             filename = unidecode(model_profile.filename_url) if romanize else model_profile.filename_url
         else:
@@ -314,7 +323,8 @@ def restart_space(repo_id: str, factory_reboot: bool):
 
 
 def extract_exif_data(image):
-    if image is None: return ""
+    if image is None:
+        return ""
 
     try:
         metadata_keys = ['parameters', 'metadata', 'prompt', 'Comment']
@@ -410,6 +420,37 @@ def download_diffuser_repo(repo_name: str, model_type: str, revision: str = "mai
     return cached_folder
 
 
+def get_folder_size_gb(folder_path):
+    result = subprocess.run(["du", "-s", folder_path], capture_output=True, text=True)
+
+    total_size_kb = int(result.stdout.split()[0])
+    total_size_gb = total_size_kb / (1024 ** 2)
+
+    return total_size_gb
+
+
+def get_used_storage_gb():
+    try:
+        used_gb = get_folder_size_gb(STORAGE_ROOT)
+        print(f"Used Storage: {used_gb:.2f} GB")
+    except Exception as e:
+        used_gb = 999
+        print(f"Error while retrieving the used storage: {e}.")
+
+    return used_gb
+
+
+def delete_model(removal_candidate):
+    print(f"Removing: {removal_candidate}")
+
+    if os.path.exists(removal_candidate):
+        os.remove(removal_candidate)
+    else:
+        diffusers_model = f"{CACHE_HF}{DIRECTORY_MODELS}--{removal_candidate.replace('/', '--')}"
+        if os.path.isdir(diffusers_model):
+            shutil.rmtree(diffusers_model)
+
+
 def progress_step_bar(step, total):
     # Calculate the percentage for the progress bar width
     percentage = min(100, ((step / total) * 100))