github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

add ability to filter list_models on list of base models

Browse Source
This commit is contained in:
Lincoln Stein
2023-07-10 21:59:32 -04:00
parent 8e42502dfd
commit 23d8a2777e
11 changed files with 341 additions and 337 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/app/api/routers/models.py
View File

@@ -32,7 +32,7 @@ class ModelsList(BaseModel):
responses={200: {"model": ModelsList }},
)
async def list_models(
base_models: Optional[List[Union[BaseModelType,None]]] = Query(default=None, description="Base models to include"),
base_models: Optional[List[BaseModelType]] = Query(default=None, description="Base models to include"),
model_type: Optional[ModelType] = Query(default=None, description="The type of model to get"),
) -> ModelsList:
"""Gets a list of models"""