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(
563
563
564
564
params = f"{ params } { deployment_params } " .strip ()
565
565
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
+ )
576
573
577
574
if params :
578
575
env_var .update ({"PARAMS" : params })
You can’t perform that action at this time.
0 commit comments