diff --git a/components/google-cloud/RELEASE.md b/components/google-cloud/RELEASE.md
index aa691855efc..0c3a04d13af 100644
--- a/components/google-cloud/RELEASE.md
+++ b/components/google-cloud/RELEASE.md
@@ -1,4 +1,6 @@
 ## Upcoming release
+* Fix Gemini batch prediction support to `v1.model_evaluation.autosxs_pipeline` after output schema change.
+
 ## Release 2.16.1
 * Fix to model batch explanation component for Structured Data pipelines; image bump.
 * Add dynamic support for boot_disk_type, boot_disk_size in `preview.custom_job.utils.create_custom_training_job_from_component`.
diff --git a/components/google-cloud/google_cloud_pipeline_components/_implementation/llm/generated/refined_image_versions.py b/components/google-cloud/google_cloud_pipeline_components/_implementation/llm/generated/refined_image_versions.py
index 7c74a59c603..9fb8dc56ff9 100644
--- a/components/google-cloud/google_cloud_pipeline_components/_implementation/llm/generated/refined_image_versions.py
+++ b/components/google-cloud/google_cloud_pipeline_components/_implementation/llm/generated/refined_image_versions.py
@@ -17,4 +17,4 @@
 DO NOT EDIT - This file is generated, manual changes will be overridden.
 """
 
-IMAGE_TAG = '20240623_1707'
+IMAGE_TAG = '20240818_1707'