Use a copy of ModelWarmupOptions for model warmup.

PiperOrigin-RevId: 764483955

Author: yimingz-a

tensorflow_serving/servables/tensorflow/saved_model_bundle_source_adapter.cc

@@ -62,15 +62,13 @@ SavedModelBundleSourceAdapter::GetServableCreator(
       MaybePublishMLMDStreamz(path, metadata.servable_id.name,
                               metadata.servable_id.version);
       if (bundle_factory->config().enable_model_warmup()) {
-        bundle_factory->mutable_config()
-            .mutable_model_warmup_options()
-            ->set_model_name(metadata.servable_id.name);
-        bundle_factory->mutable_config()
-            .mutable_model_warmup_options()
-            ->set_model_version(metadata.servable_id.version);
-        return RunSavedModelWarmup(
-            bundle_factory->config().model_warmup_options(),
-            GetRunOptions(bundle_factory->config()), path, bundle->get());
+        ModelWarmupOptions warmup_options =
+            bundle_factory->config().model_warmup_options();
+        warmup_options.set_model_name(metadata.servable_id.name);
+        warmup_options.set_model_version(metadata.servable_id.version);
+        return RunSavedModelWarmup(warmup_options,
+                                   GetRunOptions(bundle_factory->config()),
+                                   path, bundle->get());
       }
       return absl::OkStatus();
     };

tensorflow_serving/servables/tensorflow/tfrt_saved_model_factory.cc

@@ -317,11 +317,11 @@ absl::Status TfrtSavedModelFactory::CreateTfrtSavedModelWithMetadata(
       down_cast<TfrtSavedModelServable*>(servable->get());
 
   if (config().enable_model_warmup()) {
-    auto* warmup_options = mutable_config().mutable_model_warmup_options();
-    warmup_options->set_model_name(metadata.servable_id.name);
-    warmup_options->set_model_version(metadata.servable_id.version);
+    ModelWarmupOptions warmup_options = config().model_warmup_options();
+    warmup_options.set_model_name(metadata.servable_id.name);
+    warmup_options.set_model_version(metadata.servable_id.version);
     TF_RETURN_IF_ERROR(RunSavedModelWarmup(
-        *warmup_options, path, config().lazy_init_threshold(),
+        warmup_options, path, config().lazy_init_threshold(),
         config().skip_warmup_requests_if_initialized(),
         &tfrt_servable->saved_model()));
     if (config().freeze_after_init()) {