Upload 3 files

README.md

@@ -4,7 +4,7 @@ emoji: 🏆😻
 colorFrom: red
 colorTo: pink
 sdk: gradio
-sdk_version: 5.6.0
+sdk_version: 5.20.0
 app_file: app.py
 pinned: true
 license: mit

modutils.py

@@ -19,6 +19,9 @@ from unidecode import unidecode
 import copy
 from datetime import datetime, timezone, timedelta
 FILENAME_TIMEZONE = timezone(timedelta(hours=9)) # JST
+import torch
+from safetensors.torch import load_file
+import gc
 
 
 from env import (HF_LORA_PRIVATE_REPOS1, HF_LORA_PRIVATE_REPOS2,
@@ -1628,6 +1631,69 @@ def get_model_pipeline(repo_id: str):
         return default
 
 
+MODEL_TYPE_KEY = {
+    "model.diffusion_model.output_blocks.1.1.norm.bias": "SDXL",
+    "model.diffusion_model.input_blocks.11.0.out_layers.3.weight": "SD 1.5",
+    "double_blocks.0.img_attn.norm.key_norm.scale": "FLUX",
+    "model.diffusion_model.double_blocks.0.img_attn.norm.key_norm.scale": "FLUX",
+    "model.diffusion_model.joint_blocks.9.x_block.attn.ln_k.weight": "SD 3.5",
+}
+
+
+def safe_clean(path: str):
+    try:
+        if Path(path).exists():
+            if Path(path).is_dir(): shutil.rmtree(str(Path(path)))
+            else: Path(path).unlink()
+            print(f"Deleted: {path}")
+        else: print(f"File not found: {path}")
+    except Exception as e:
+        print(f"Failed to delete: {path} {e}")
+
+
+def read_safetensors_key(path: str):
+    try:
+        keys = []
+        state_dict = load_file(str(Path(path)))
+        for k in list(state_dict.keys()):
+            keys.append(k)
+            state_dict.pop(k)
+    except Exception as e:
+        print(e)
+    finally:
+        del state_dict
+        torch.cuda.empty_cache()
+        gc.collect()
+        return keys
+
+
+def get_model_type_from_key(path: str):
+    default = "SDXL"
+    try:
+        keys = read_safetensors_key(path)
+        for k, v in MODEL_TYPE_KEY.items():
+            if k in set(keys):
+                print(f"Model type is {v}.")
+                return v
+        print("Model type could not be identified.")
+    except Exception:
+        return default
+    return default
+
+
+def download_link_model(url: str, localdir: str):
+    try:
+        new_file = None
+        new_file = get_download_file(localdir, url, CIVITAI_API_KEY)
+        if not new_file or Path(new_file).suffix.lower() not in set([".safetensors", ".ckpt", ".bin", ".sft"]):
+            if Path(new_file).exists(): Path(new_file).unlink()
+            raise gr.Error(f"Safetensors file not found: {url}")
+        model_type = get_model_type_from_key(new_file)
+        return new_file, model_type
+    except Exception as e:
+        raise gr.Error(f"Failed to load single model file: {url} {e}")
+
+
 EXAMPLES_GUI = [
     [
         "1girl, souryuu asuka langley, neon genesis evangelion, plugsuit, pilot suit, red bodysuit, sitting, crossing legs, black eye patch, cat hat, throne, symmetrical, looking down, from bottom, looking at viewer, outdoors, masterpiece, best quality, very aesthetic, absurdres",
@@ -1640,7 +1706,7 @@ EXAMPLES_GUI = [
         "Euler",
         1152,
         896,
-        "votepurchase/animagine-xl-3.1",
+        "cagliostrolab/animagine-xl-4.0",
     ],
     [
         "solo, princess Zelda OOT, score_9, score_8_up, score_8, medium breasts, cute, eyelashes, cute small face, long hair, crown braid, hairclip, pointy ears, soft curvy body, looking at viewer, smile, blush, white dress, medium body, (((holding the Master Sword))), standing, deep forest in the background",
@@ -1679,7 +1745,7 @@ EXAMPLES_GUI = [
         "Euler",
         1024,
         1024,
-        "Raelina/Raemu-XL-V4",
+        "Raelina/Raemu-XL-V5",
     ],
     [
         "yoshida yuuko, machikado mazoku, 1girl, solo, demon horns,horns, school uniform, long hair, open mouth, skirt, demon girl, ahoge, shiny, shiny hair, anime artwork",
@@ -1692,7 +1758,7 @@ EXAMPLES_GUI = [
         "Euler",
         1024,
         1024,
-        "cagliostrolab/animagine-xl-3.1",
+        "cagliostrolab/animagine-xl-4.0",
     ],
 ]
 

requirements.txt

@@ -1,6 +1,6 @@
-torch
+torch==2.4.0
 git+https://github.com/huggingface/diffusers.git
-git+https://github.com/huggingface/transformers.git
+transformers==4.48.3
 git+https://github.com/huggingface/peft.git
 git+https://github.com/huggingface/accelerate.git
 sentencepiece