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

Added check for missing preview uri.

Browse Source
This commit is contained in:
Christian Bastian
2024-09-07 17:08:44 -04:00
parent ee9489b758
commit 84b1e60df0
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
__init__.py
View File

@@ -417,6 +417,8 @@ def get_auto_thumbnail_format(original_format):
@server.PromptServer.instance.routes.get("/model-manager/preview/get") @server.PromptServer.instance.routes.get("/model-manager/preview/get")
async def get_model_preview(request): async def get_model_preview(request):
uri = request.query.get("uri") uri = request.query.get("uri")
if uri is None:
raise ValueError("Missing uri path!")
quality = 75 quality = 75
response_image_format = request.query.get("image-format", None) response_image_format = request.query.get("image-format", None)
if isinstance(response_image_format, str): if isinstance(response_image_format, str):