fix: LLM - Fixed filter in list_tuned_model_names

PiperOrigin-RevId: 555359431
googleapis · Aug 10, 2023 · 57806fb · 57806fb
1 parent 40f3e41
commit 57806fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vertexai/language_models/_language_models.py b/vertexai/language_models/_language_models.py
@@ -1083,7 +1083,7 @@ def _get_tuned_models_dir_uri(model_id: str) -> str:
 
 def _list_tuned_model_names(model_id: str) -> List[str]:
     tuned_models = aiplatform.Model.list(
-        filter=f'labels.{_TUNING_BASE_MODEL_ID_LABEL_KEY}="{model_id}"',
+        filter=f'labels.{_TUNING_BASE_MODEL_ID_LABEL_KEY}="{model_id.replace("@", "-")}"',
         # TODO(b/275444096): Remove the explicit location once models are deployed to the user's selected location
         location=_TUNED_MODEL_LOCATION,
     )