test2

README.md

@@ -13,6 +13,7 @@ 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,11 +199,6 @@ 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:
@@ -228,34 +223,6 @@ 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:
@@ -2288,6 +2255,7 @@ 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"
+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"
 
 LOAD_DIFFUSERS_FORMAT_MODEL = [
     'TestOrganizationPleaseIgnore/potato_quality_anime_plzwork_sdxl',

env.py

@@ -30,8 +30,6 @@ LOAD_DIFFUSERS_FORMAT_MODEL = [
     'BlueDancer/Artisanica_XL',
     'neta-art/neta-noob-1.0',
     'OnomaAIResearch/Illustrious-xl-early-release-v0',
-    'martineux/bismuth7-xl',
-    'martineux/perfectdeliberate8',
     'Raelina/Rae-Diffusion-XL-V2',
     'Raelina/Raemu-XL-V4',
     'Raelina/Raemu-XL-V5',
@@ -47,10 +45,6 @@ LOAD_DIFFUSERS_FORMAT_MODEL = [
     'Raelina/Raehoshi-illust-XL-7',
     'Raelina/Raehoshi-illust-XL-7.1',
     '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',
     'sayakpaul/FLUX.1-merged',

packages.txt

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

requirements.txt

@@ -3,8 +3,10 @@ diffusers
 transformers
 accelerate
 huggingface_hub
-spaces
-torch==2.8.0
+hf_transfer
+hf_xet
+torch==2.5.1
+torchvision
 numpy<2
 gdown
 opencv-python
@@ -17,6 +19,5 @@ rapidfuzz
 sentencepiece
 unidecode
 matplotlib-inline
-mediapipe==0.10.5
-einops
-# pydantic==2.10.6
+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

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
 
-    wget_command = (
-        f'wget -c -nv '
-        f'-O "{os.path.join(dir_, filename_base)}" "{redirect_url}"'
+    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}"'
     )
-    r_code = os.system(wget_command)  # noqa
+    r_code = os.system(aria2_command)  # noqa
 
     # if r_code != 0:
     #     raise RuntimeError(f"Failed to download file: {filename_base}. Error code: {r_code}")
@@ -293,32 +293,27 @@ 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 wget.
-    If an API key limit is reached, generating a new API key and using it can fix the issue. no
+    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.
     """
     output_path = None
-    
+
     url_dl = url + f"?token={civitai_api_key}"
-    
     if not civ_filename:
-        wget_command = (
-            f'wget -c -nv '
-            f'-P "{dir_}" "{url_dl}"'
-        )
-        os.system(wget_command)
-    
+        aria2_command = f'aria2c -c -x 1 -s 1 -d "{dir_}" "{url_dl}"'
+        os.system(aria2_command)
     else:
         output_path = os.path.join(dir_, civ_filename)
-    
         if not os.path.exists(output_path):
-            wget_command = (
-                f'wget -c -nv '
-                f'-O "{output_path}" "{url_dl}"'
+            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}"'
             )
-            os.system(wget_command)
-    
+            os.system(aria2_command)
+
     return output_path
 
+
 def drive_down(url, dir_):
     import gdown
 
@@ -359,16 +354,10 @@ def hf_down(url, dir_, hf_token, romanize):
         url = url.replace("/blob/", "/resolve/")
 
     if hf_token:
-        os.system(
-            f'wget -c -nv '
-            f'--header="Authorization: Bearer {hf_token}" '
-            f'-O "{os.path.join(dir_, filename)}" "{url}"'
-        )
+        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}")
     else:
-        os.system(
-            f'wget -c -nv '
-            f'-O "{os.path.join(dir_, filename)}" "{url}"'
-        )
+        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}")
 
     return output_path
 
@@ -381,8 +370,7 @@ 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." in url:
-        url = url.replace("civitai.red", "civitai.com")
+    elif "civitai.com" in url:
         if not civitai_api_key:
             msg = "You need an API key to download Civitai models."
             print(f"\033[91m{msg}\033[0m")
@@ -405,10 +393,7 @@ 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'wget -c -nv '
-            f'-P "{directory}" "{url}"'
-        )
+        os.system(f"aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 -d {directory} {url}")
 
     return downloaded_file_path