diff --git a/components/google-cloud/google_cloud_pipeline_components/_implementation/model_evaluation/endpoint_batch_predict/component.py b/components/google-cloud/google_cloud_pipeline_components/_implementation/model_evaluation/endpoint_batch_predict/component.py
index e470b462bca..21a3ca5718c 100644
--- a/components/google-cloud/google_cloud_pipeline_components/_implementation/model_evaluation/endpoint_batch_predict/component.py
+++ b/components/google-cloud/google_cloud_pipeline_components/_implementation/model_evaluation/endpoint_batch_predict/component.py
@@ -143,7 +143,7 @@ def evaluation_llm_endpoint_batch_predict_pipeline_graph_component(
     model_parameters: Optional[Dict[str, Union[int, float]]] = {},
     endpoint_id: Optional[str] = '',
     publisher_model: Optional[str] = '',
-    qms_override: Optional[str] = None,
+    qms_override: Optional[Dict[str, Union[int, float]]] = None,
     display_name: str = 'endpoint_batch_predict',
     machine_type: str = 'e2-highmem-16',
     service_account: str = '',
@@ -214,7 +214,7 @@ def evaluation_llm_endpoint_batch_predict_pipeline_graph_component(
       ).output,
       endpoint_id=endpoint_id,
       publisher_model=publisher_model,
-      qms_override=qms_override,
+      qms_override=add_json_escape_parameters(parameters=qms_override).output,
       display_name=display_name,
       machine_type=machine_type,
       service_account=service_account,