From 4a059f3ae050c28a2fa12aa0897d7c8f634c0e6c Mon Sep 17 00:00:00 2001 From: Dmitrii Cherkasov Date: Wed, 17 Sep 2025 10:55:40 -0700 Subject: [PATCH] Fixes an issue where the --served-model-name attribute was being added by default to all deployments. --- ads/aqua/modeldeployment/deployment.py | 17 +++++++---------- 1 file changed, 7 insertions(+), 10 deletions(-) diff --git a/ads/aqua/modeldeployment/deployment.py b/ads/aqua/modeldeployment/deployment.py index e6ba00a93..72d655391 100644 --- a/ads/aqua/modeldeployment/deployment.py +++ b/ads/aqua/modeldeployment/deployment.py @@ -563,16 +563,13 @@ def _create( params = f"{params} {deployment_params}".strip() - if create_deployment_details.model_name: - # Replace existing --served-model-name argument if present, otherwise add it - if "--served-model-name" in params: - params = re.sub( - r"--served-model-name\s+\S+", - f"--served-model-name {create_deployment_details.model_name}", - params, - ) - else: - params += f" --served-model-name {create_deployment_details.model_name}" + if create_deployment_details.model_name and "--served-model-name" in params: + # Replace existing --served-model-name argument with custom name provided by user + params = re.sub( + r"--served-model-name\s+\S+", + f"--served-model-name {create_deployment_details.model_name}", + params, + ) if params: env_var.update({"PARAMS": params})