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})