Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

allow metadata-less models to be used for GET metadata endpoint

Browse Source
This commit is contained in:
maryhipp 2024-02-22 09:33:38 -05:00 committed by psychedelicious
parent 8d243b1fca
commit 091bf9220b
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
invokeai/app/api/routers/model_manager.py
View File

@ -202,7 +202,7 @@ async def list_model_summary(
@model_manager_router.get( @model_manager_router.get(
"/meta/i/{key}", "/i/{key}/meta",
operation_id="get_model_metadata", operation_id="get_model_metadata",
responses={ responses={
200: { 200: {
@ -210,7 +210,6 @@ async def list_model_summary(
"content": {"application/json": {"example": example_model_metadata}}, "content": {"application/json": {"example": example_model_metadata}},
}, },
400: {"description": "Bad request"}, 400: {"description": "Bad request"},
404: {"description": "No metadata available"},
}, },
) )
async def get_model_metadata( async def get_model_metadata(
@ -219,8 +218,7 @@ async def get_model_metadata(
"""Get a model metadata object.""" """Get a model metadata object."""
record_store = ApiDependencies.invoker.services.model_manager.store record_store = ApiDependencies.invoker.services.model_manager.store
result: Optional[AnyModelRepoMetadata] = record_store.get_metadata(key) result: Optional[AnyModelRepoMetadata] = record_store.get_metadata(key)
if not result:
raise HTTPException(status_code=404, detail="No metadata for a model with this key")
return result return result