From af1c1ab51f088d7537ba004f6b5379c95aa3806d Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Mon, 17 Jul 2023 22:33:05 -0400 Subject: [PATCH] importing an unrecognized model now gives "Unsupported Media Type" error --- invokeai/app/api/routers/models.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/invokeai/app/api/routers/models.py b/invokeai/app/api/routers/models.py index 33866caa26..c2741fb242 100644 --- a/invokeai/app/api/routers/models.py +++ b/invokeai/app/api/routers/models.py @@ -128,6 +128,7 @@ async def update_model( responses= { 201: {"description" : "The model imported successfully"}, 404: {"description" : "The model could not be found"}, + 415: {"description" : "Unrecognized file/folder format"}, 424: {"description" : "The model appeared to import successfully, but could not be found in the model manager"}, 409: {"description" : "There is already a model corresponding to this path or repo_id"}, }, @@ -154,7 +155,7 @@ async def import_model( if not info: logger.error("Import failed") - raise HTTPException(status_code=424) + raise HTTPException(status_code=415) logger.info(f'Successfully imported {location}, got {info}') model_raw = ApiDependencies.invoker.services.model_manager.list_model(