feat(components): Use 64 v3 TPUs for llm pipelines

PiperOrigin-RevId: 568282755
kubeflow · Sep 25, 2023 · 45fe8e8 · 45fe8e8
1 parent 173be83
commit 45fe8e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...nents/google-cloud/google_cloud_pipeline_components/_implementation/llm/function_based.py b/...nents/google-cloud/google_cloud_pipeline_components/_implementation/llm/function_based.py
@@ -57,7 +57,7 @@ def resolve_machine_spec(
     return outputs(
         machine_type='cloud-tpu',
         accelerator_type='TPU_V3',
-        accelerator_count=32,
+        accelerator_count=64,
     )
   elif location in gpu_regions:
     return outputs(