1580158
/
Stable-Diffusion-Webui-Civitai-Helper
mirror of https://github.com/butaixianran/Stable-Diffusion-Webui-Civitai-Helper.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #266 from man90es/api-key-error-message 

Add an error message about API key
pull/273/head
butaixianran 2024-01-07 22:56:08 +08:00 committed by GitHub
parent 3671729087 972ac08904
commit bba0b515a1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/ch_lib/downloader.py
View File

@ -33,8 +33,10 @@ def dl(url, folder, filename, filepath):
# first request for header # first request for header
rh = requests.get(url, stream=True, verify=False, headers=util.def_headers, proxies=util.proxies) rh = requests.get(url, stream=True, verify=False, headers=util.def_headers, proxies=util.proxies)
# get file size # get file size
total_size = 0 total_size = int(rh.headers.get('Content-Length', -1))
total_size = int(rh.headers['Content-Length']) if (total_size < 0):
util.printD('This model requires an API key to download. More info: https://github.com/butaixianran/Stable-Diffusion-Webui-Civitai-Helper#civitai-api-key')
return
util.printD(f"File size: {total_size}") util.printD(f"File size: {total_size}")
# if file_path is empty, need to get file name from download url's header # if file_path is empty, need to get file name from download url's header
@ -121,4 +123,3 @@ def dl(url, folder, filename, filepath):
os.rename(dl_file_path, file_path) os.rename(dl_file_path, file_path)
util.printD(f"File Downloaded to: {file_path}") util.printD(f"File Downloaded to: {file_path}")
return file_path return file_path