fix function arguments for get_model_list

theroyallab · Sep 7, 2024 · d57a3b4 · d57a3b4
1 parent d34756d
commit d57a3b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/endpoints/core/router.py b/endpoints/core/router.py
@@ -103,7 +103,7 @@ async def list_draft_models(request: Request) -> ModelList:
 
         models = get_model_list(draft_model_path.resolve())
     else:
-        models = await get_current_model_list(is_draft=True)
+        models = await get_current_model_list(model_type="draft")
 
     return models