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

Added a bit more safety checks before saving url.

Browse Source
This commit is contained in:
Christian Bastian
2024-09-12 13:41:29 -04:00
parent 15946599fb
commit c87183f98b
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
__init__.py
View File

@@ -1206,7 +1206,7 @@ async def get_model_web_url(request):
result["alert"] = "Unable to find model info!" result["alert"] = "Unable to find model info!"
return web.json_response(result) return web.json_response(result)
web_url = ModelInfo.get_url(model_info) web_url = ModelInfo.get_url(model_info)
if web_url != "": if web_url != "" and web_url is not None:
save_web_url(url_path, web_url) save_web_url(url_path, web_url)
result["success"] = True result["success"] = True
@@ -1262,7 +1262,8 @@ async def download_model(request):
# save url # save url
url_file_path = os.path.splitext(file_name)[0] + ".url" url_file_path = os.path.splitext(file_name)[0] + ".url"
url = ModelInfo.get_url(url) url = ModelInfo.get_url(url)
save_web_url(url_file_path, url) if url != "" and url is not None:
save_web_url(url_file_path, url)
# save image as model preview # save image as model preview
image = formdata.get("image") image = formdata.get("image")