John6666 commited on
Commit
1b11b90
·
verified ·
1 Parent(s): dfa3eb1

Upload 2 files

Browse files
Files changed (2) hide show
  1. dc.py +2 -2
  2. modutils.py +2 -2
dc.py CHANGED
@@ -197,8 +197,8 @@ def download_things(directory, url, hf_token="", civitai_api_key=""):
197
  if "?" in url:
198
  url = url.split("?")[0]
199
  if civitai_api_key:
200
- user_header = f'"Authorization: Bearer {civitai_api_key}"'
201
- os.system(f"aria2c --console-log-level=error --summary-interval=10 --header={user_header} -c -x 16 -k 1M -s 16 -d {directory} {url}")
202
  else:
203
  print("\033[91mYou need an API key to download Civitai models.\033[0m")
204
  else:
 
197
  if "?" in url:
198
  url = url.split("?")[0]
199
  if civitai_api_key:
200
+ url = url + f"?token={civitai_api_key}"
201
+ os.system(f"aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 -d {directory} {url}")
202
  else:
203
  print("\033[91mYou need an API key to download Civitai models.\033[0m")
204
  else:
modutils.py CHANGED
@@ -82,8 +82,8 @@ def download_things(directory, url, hf_token="", civitai_api_key=""):
82
  if "?" in url:
83
  url = url.split("?")[0]
84
  if civitai_api_key:
85
- user_header = f'"Authorization: Bearer {civitai_api_key}"'
86
- os.system(f"aria2c --console-log-level=error --summary-interval=10 --header={user_header} -c -x 16 -k 1M -s 16 -d {directory} {url}")
87
  else:
88
  print("\033[91mYou need an API key to download Civitai models.\033[0m")
89
  else:
 
82
  if "?" in url:
83
  url = url.split("?")[0]
84
  if civitai_api_key:
85
+ url = url + f"?token={civitai_api_key}"
86
+ os.system(f"aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 -d {directory} {url}")
87
  else:
88
  print("\033[91mYou need an API key to download Civitai models.\033[0m")
89
  else: