Upload 23 files

README.md

@@ -13,7 +13,6 @@ 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
 ---
 
 ## Using this Space programmatically

app.py

@@ -199,6 +199,11 @@ class GuiSD:
         # Avoid duplicate downloads
         self.active_downloads = set()
         self.download_lock = threading.Lock()
+
+        # Anti-abuse: track new model requests.
+        self.used_models = []
+        self.new_model_history = []
+
     def update_storage_models(self, storage_floor_gb=24, required_inventory_for_purge=3):
         while get_used_storage_gb() > storage_floor_gb:
             if len(self.inventory) < required_inventory_for_purge:
@@ -223,6 +228,34 @@ class GuiSD:
         print(self.inventory)
 
     def load_new_model(self, model_name, vae_model, task, controlnet_model, progress=gr.Progress(track_tqdm=True)):
+
+        if model_name != model_list[0]:
+            # --- Anti-Abuse Check Start ---
+            if model_name in self.used_models:
+                # Move to the end to mark as the most recently used.
+                self.used_models.remove(model_name)
+                self.used_models.append(model_name)
+            else:
+                current_time = datetime.now()
+                # Retain history of new model requests from the last 20 minutes.
+                self.new_model_history = [
+                    t for t in self.new_model_history
+                    if (current_time - t).total_seconds() < 1200
+                ]
+
+                # Allow a maximum of 5 new model requests per 20 minutes.
+                if len(self.new_model_history) >= 5:
+                    yield "Rate limit exceeded: Too many new models requested."
+                    raise gr.Error("Too many new models requested. Please reuse your previously loaded models or wait a few minutes before trying new ones.")
+
+                self.new_model_history.append(current_time)
+                self.used_models.append(model_name)
+
+                # Cap the reuse list to the 5 most recent models.
+                if len(self.used_models) > 5:
+                    self.used_models.pop(0)
+            # --- Anti-Abuse Check End ---
+
         lock_key = model_name
 
         while True:
@@ -2255,7 +2288,6 @@ with gr.Blocks(theme=args.theme, elem_id="main", fill_width=True, fill_height=Fa
     gr.api(generate_image, api_name="generate_image", show_api=True, queue=True, concurrency_id="gpu")
     gr.api(generate_image_stream, api_name="generate_image_stream", show_api=True, queue=True, concurrency_id="gpu")
 
-    gr.LoginButton()
     gr.DuplicateButton(value="Duplicate Space for private use (This demo does not work on CPU. Requires GPU Space)")
 
 

constants.py

@@ -19,7 +19,7 @@ DOWNLOAD_MODEL = "https://huggingface.co/zuv0/test/resolve/main/milkyWonderland_
 DOWNLOAD_VAE = "https://huggingface.co/Anzhc/Anzhcs-VAEs/resolve/main/SDXL%20Anime%20VAE%20Dec-only%20B3.safetensors, 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"
+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"
 
 LOAD_DIFFUSERS_FORMAT_MODEL = [
     'TestOrganizationPleaseIgnore/potato_quality_anime_plzwork_sdxl',

env.py

@@ -49,6 +49,7 @@ LOAD_DIFFUSERS_FORMAT_MODEL = [
     'Raelina/Raehoshi-illust-XL-8',
     'Raelina/Raehoshi-illust-XL-8.1',
     'Raelina/Raehoshi-illust-XL-9',
+    'Raelina/Raehoshi-illust-XL-9.1',
     'Raelina/Raehoshi-illust-vpred',
     'camenduru/FLUX.1-dev-diffusers',
     'black-forest-labs/FLUX.1-schnell',

packages.txt

@@ -1 +1,2 @@
-git-lfs aria2 ffmpeg
+git-lfs
+ffmpeg

requirements.txt

@@ -3,10 +3,8 @@ diffusers
 transformers
 accelerate
 huggingface_hub
-#hf_transfer
-hf_xet
-torch==2.5.1
-torchvision
+spaces
+torch==2.8.0
 numpy<2
 gdown
 opencv-python
@@ -20,5 +18,5 @@ sentencepiece
 unidecode
 matplotlib-inline
 mediapipe==0.10.5
-https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.4.11/flash_attn-2.8.3+cu124torch2.5-cp310-cp310-linux_x86_64.whl
-#pydantic==2.10.6
+einops
+# pydantic==2.10.6

utils.py

@@ -275,11 +275,11 @@ def civ_redirect_down(url, dir_, civitai_api_key, romanize, alternative_name):
     elif os.path.exists(os.path.join(dir_, filename_base)):
         return os.path.join(dir_, filename_base), filename_base
 
-    aria2_command = (
-        f'aria2c --console-log-level=error --summary-interval=10 -c -x 16 '
-        f'-k 1M -s 16 -d "{dir_}" -o "{filename_base}" "{redirect_url}"'
+    wget_command = (
+        f'wget -c -nv '
+        f'-O "{os.path.join(dir_, filename_base)}" "{redirect_url}"'
     )
-    r_code = os.system(aria2_command)  # noqa
+    r_code = os.system(wget_command)  # noqa
 
     # if r_code != 0:
     #     raise RuntimeError(f"Failed to download file: {filename_base}. Error code: {r_code}")
@@ -293,27 +293,32 @@ def civ_redirect_down(url, dir_, civitai_api_key, romanize, alternative_name):
 
 def civ_api_down(url, dir_, civitai_api_key, civ_filename):
     """
-    This method is susceptible to being blocked because it generates a lot of temp redirect links with aria2c.
-    If an API key limit is reached, generating a new API key and using it can fix the issue.
+    This method is susceptible to being blocked because it generates a lot of temp redirect links with wget.
+    If an API key limit is reached, generating a new API key and using it can fix the issue. no
     """
     output_path = None
-
+    
     url_dl = url + f"?token={civitai_api_key}"
+    
     if not civ_filename:
-        aria2_command = f'aria2c -c -x 1 -s 1 -d "{dir_}" "{url_dl}"'
-        os.system(aria2_command)
+        wget_command = (
+            f'wget -c -nv '
+            f'-P "{dir_}" "{url_dl}"'
+        )
+        os.system(wget_command)
+    
     else:
         output_path = os.path.join(dir_, civ_filename)
+    
         if not os.path.exists(output_path):
-            aria2_command = (
-                f'aria2c --console-log-level=error --summary-interval=10 -c -x 16 '
-                f'-k 1M -s 16 -d "{dir_}" -o "{civ_filename}" "{url_dl}"'
+            wget_command = (
+                f'wget -c -nv '
+                f'-O "{output_path}" "{url_dl}"'
             )
-            os.system(aria2_command)
-
+            os.system(wget_command)
+    
     return output_path
 
-
 def drive_down(url, dir_):
     import gdown
 
@@ -354,10 +359,16 @@ def hf_down(url, dir_, hf_token, romanize):
         url = url.replace("/blob/", "/resolve/")
 
     if hf_token:
-        user_header = f'"Authorization: Bearer {hf_token}"'
-        os.system(f"aria2c --console-log-level=error --summary-interval=10 --header={user_header} -c -x 16 -k 1M -s 16 {url} -d {dir_}  -o {filename}")
+        os.system(
+            f'wget -c -nv '
+            f'--header="Authorization: Bearer {hf_token}" '
+            f'-O "{os.path.join(dir_, filename)}" "{url}"'
+        )
     else:
-        os.system(f"aria2c --optimize-concurrent-downloads --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 {url} -d {dir_}  -o {filename}")
+        os.system(
+            f'wget -c -nv '
+            f'-O "{os.path.join(dir_, filename)}" "{url}"'
+        )
 
     return output_path
 
@@ -370,7 +381,8 @@ def download_things(directory, url, hf_token="", civitai_api_key="", romanize=Fa
         downloaded_file_path = drive_down(url, directory)
     elif "huggingface.co" in url:
         downloaded_file_path = hf_down(url, directory, hf_token, romanize)
-    elif "civitai.com" in url:
+    elif "civitai." in url:
+        url = url.replace("civitai.red", "civitai.com")
         if not civitai_api_key:
             msg = "You need an API key to download Civitai models."
             print(f"\033[91m{msg}\033[0m")
@@ -393,7 +405,10 @@ def download_things(directory, url, hf_token="", civitai_api_key="", romanize=Fa
             gr.Warning(msg)
             downloaded_file_path = civ_api_down(url, directory, civitai_api_key, civ_filename)
     else:
-        os.system(f"aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 -d {directory} {url}")
+        os.system(
+            f'wget -c -nv '
+            f'-P "{directory}" "{url}"'
+        )
 
     return downloaded_file_path