Skip to content

Commit a40168f

Browse files
authored
Fixes an issue where the --served-model-name attribute was being added by default to all deployments. (#1271)
1 parent ed095fd commit a40168f

File tree

1 file changed

+7
-10
lines changed

1 file changed

+7
-10
lines changed

ads/aqua/modeldeployment/deployment.py

Lines changed: 7 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -563,16 +563,13 @@ def _create(
563563

564564
params = f"{params} {deployment_params}".strip()
565565

566-
if create_deployment_details.model_name:
567-
# Replace existing --served-model-name argument if present, otherwise add it
568-
if "--served-model-name" in params:
569-
params = re.sub(
570-
r"--served-model-name\s+\S+",
571-
f"--served-model-name {create_deployment_details.model_name}",
572-
params,
573-
)
574-
else:
575-
params += f" --served-model-name {create_deployment_details.model_name}"
566+
if create_deployment_details.model_name and "--served-model-name" in params:
567+
# Replace existing --served-model-name argument with custom name provided by user
568+
params = re.sub(
569+
r"--served-model-name\s+\S+",
570+
f"--served-model-name {create_deployment_details.model_name}",
571+
params,
572+
)
576573

577574
if params:
578575
env_var.update({"PARAMS": params})

0 commit comments

Comments
 (0)