TBNilles/ComfyUI-Model-Manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove debug print().

Browse Source
This commit is contained in:
Christian Bastian
2024-02-13 15:40:11 -05:00
parent f810f68bbe
commit 8fbbff2175
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
__init__.py
View File

@@ -321,7 +321,6 @@ def download_file(url, filename, overwrite):
def_headers["Authorization"] = f"Bearer {api_key}" def_headers["Authorization"] = f"Bearer {api_key}"
rh = requests.get(url=url, stream=True, verify=False, headers=def_headers, proxies=None, allow_redirects=False) rh = requests.get(url=url, stream=True, verify=False, headers=def_headers, proxies=None, allow_redirects=False)
print(rh.status_code)
if not rh.ok: if not rh.ok:
raise Exception("Unable to download") raise Exception("Unable to download")
@@ -333,12 +332,10 @@ def download_file(url, filename, overwrite):
headers["User-Agent"] = def_headers["User-Agent"] headers["User-Agent"] = def_headers["User-Agent"]
r = requests.get(url=url, stream=True, verify=False, headers=headers, proxies=None, allow_redirects=False) r = requests.get(url=url, stream=True, verify=False, headers=headers, proxies=None, allow_redirects=False)
print(r.status_code)
if rh.status_code == 307 and r.status_code == 307: if rh.status_code == 307 and r.status_code == 307:
# Civitai redirect # Civitai redirect
redirect_url = r.content.decode("utf-8") redirect_url = r.content.decode("utf-8")
if not redirect_url.startswith("http"): if not redirect_url.startswith("http"):
print(redirect_url)
# Civitai requires login (NSFW or user-required) # Civitai requires login (NSFW or user-required)
# TODO: inform user WHY download failed # TODO: inform user WHY download failed
raise Exception("Unable to download!") raise Exception("Unable to download!")
@@ -428,7 +425,6 @@ async def download_model(request):
download_uri = body.get("download") download_uri = body.get("download")
if download_uri is None: if download_uri is None:
result["invalid"] = "download" result["invalid"] = "download"
print(download_uri)
return web.json_response(result) return web.json_response(result)
name = body.get("name") name = body.get("name")