Upload 5 files

env.py

@@ -45,6 +45,7 @@ models = [
     "John6666/nsfw-master-flux-lora-merged-with-flux1-dev-fp16-v10-fp8-flux",
     "John6666/the-araminta-flux1a1-fp8-flux",
     "John6666/acorn-is-spinning-flux-v11-fp8-flux",
+    "John6666/stoiqo-afrodite-fluxxl-f1dalpha-fp8-flux",
     "John6666/real-horny-pro-fp8-flux",
     "John6666/centerfold-flux-v20fp8e5m2-fp8-flux",
     "John6666/jib-mix-flux-v208stephyper-fp8-flux",

flux.py

@@ -24,7 +24,6 @@ from modutils import (to_list, list_uniq, list_sub, get_lora_model_list, downloa
 
 def download_things(directory, url, hf_token="", civitai_api_key=""):
     url = url.strip()
-    
     if "drive.google.com" in url:
         original_dir = os.getcwd()
         os.chdir(directory)
@@ -44,8 +43,8 @@ def download_things(directory, url, hf_token="", civitai_api_key=""):
         if "?" in url:
             url = url.split("?")[0]
         if civitai_api_key:
-            url = url + f"?token={civitai_api_key}"
-            os.system(f"aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 -d {directory} {url}")
+            user_header = f'"Authorization: Bearer {civitai_api_key}"'
+            os.system(f"aria2c --console-log-level=error --summary-interval=10 --header={user_header} -c -x 16 -k 1M -s 16 -d {directory} {url}")
         else:
             print("\033[91mYou need an API key to download Civitai models.\033[0m")
     else:

modutils.py

@@ -82,8 +82,8 @@ def download_things(directory, url, hf_token="", civitai_api_key=""):
         if "?" in url:
             url = url.split("?")[0]
         if civitai_api_key:
-            url = url + f"?token={civitai_api_key}"
-            os.system(f"aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 -d {directory} {url}")
+            user_header = f'"Authorization: Bearer {civitai_api_key}"'
+            os.system(f"aria2c --console-log-level=error --summary-interval=10 --header={user_header} -c -x 16 -k 1M -s 16 -d {directory} {url}")
         else:
             print("\033[91mYou need an API key to download Civitai models.\033[0m")
     else:
@@ -120,7 +120,7 @@ def save_images(images: list[Image.Image], metadatas: list[str]):
         output_images = []
         for image, metadata in zip(images, metadatas):
             info = PngImagePlugin.PngInfo()
-            info.add_text("metadata", metadata)
+            info.add_text("parameters", metadata)
             savefile = f"{str(uuid.uuid4())}.png"
             image.save(savefile, "PNG", pnginfo=info)
             output_images.append(str(Path(savefile).resolve()))
@@ -802,7 +802,6 @@ def search_lora_on_civitai(query: str, allow_model: list[str] = ["Pony", "SDXL 1
     import requests
     from requests.adapters import HTTPAdapter
     from urllib3.util import Retry
-    
     user_agent = get_user_agent()
     headers = {'User-Agent': user_agent, 'content-type': 'application/json'}
     base_url = 'https://civitai.com/api/v1/models'