File tree Expand file tree Collapse file tree 1 file changed +7
-10
lines changed Expand file tree Collapse file tree 1 file changed +7
-10
lines changed Original file line number Diff line number Diff 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 })
You can’t perform that action at this time.
0 commit comments