Skip to content

Commit 541870f

Browse files
committed
fix get models bug
1 parent 9f84f71 commit 541870f

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

src/rest/rest.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -216,9 +216,13 @@ def _register_routes(self):
216216
self.router.get("/voiceclone/status")(self.get_status)
217217

218218
async def get_available_models(self):
219-
gpts = ["default"].extend(list_train_gpts().keys())
220-
sovits = ["default"].extend(list_train_sovits().keys())
221-
return {"gpts": gpts, "sovits": sovits}
219+
try:
220+
gpts = ["default"].extend(list_train_gpts().keys())
221+
sovits = ["default"].extend(list_train_sovits().keys())
222+
return {"gpts": gpts, "sovits": sovits}
223+
except Exception as e:
224+
logger.error(f"failed to get available models: {e}")
225+
raise HTTPException(status_code=HTTPStatus.INTERNAL_SERVER_ERROR, detail={"error": f"failed to get available models: {e}"})
222226

223227
async def get_status(self):
224228
if self.service is None:

0 commit comments

Comments
 (0)