set lmi config on js model optimize

Author: gwang111

src/sagemaker/serve/builder/model_builder.py

@@ -1302,6 +1302,8 @@ def _model_builder_optimize_wrapper(
         job_name = job_name or f"modelbuilderjob-{uuid.uuid4().hex}"
         if self._is_jumpstart_model_id():
             self.build(mode=self.mode, sagemaker_session=self.sagemaker_session)
+            if self.pysdk_model:
+                self.pysdk_model.set_deployment_config(instance_type=instance_type, config_name="lmi")
             input_args = self._optimize_for_jumpstart(
                 output_path=output_path,
                 instance_type=instance_type,