From 404875cab94ba69c1b92455f09859bf59ee6d512 Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Fri, 24 May 2024 02:11:17 -0700 Subject: [PATCH] fix --- backend/main.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/backend/main.py b/backend/main.py index 24442488a..fb65aa3e6 100644 --- a/backend/main.py +++ b/backend/main.py @@ -307,7 +307,10 @@ async def get_models(user=Depends(get_verified_user)): for custom_model in custom_models: if custom_model.base_model_id == None: for model in models: - if custom_model.id == model["id"]: + if ( + custom_model.id == model["id"] + or custom_model.id == model["id"].split(":")[0] + ): model["name"] = custom_model.name model["info"] = custom_model.model_dump() else: