Correct distributed_type in the accelerate test. (#121)

Co-authored-by: Will Cromar <[email protected]>
GoogleCloudPlatform · Feb 16, 2024 · 17f33c0 · 17f33c0
1 parent 7475e2a
commit 17f33c0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dags/legacy_test/tests/pytorch/nightly/common.libsonnet b/dags/legacy_test/tests/pytorch/nightly/common.libsonnet
@@ -152,7 +152,7 @@ local volumes = import 'templates/volumes.libsonnet';
         mkdir -p ~/.cache/huggingface/accelerate/
         cat > ~/.cache/huggingface/accelerate/default_config.yaml << 'HF_CONFIG_EOF'
         compute_environment: LOCAL_MACHINE
-        distributed_type: TPU
+        distributed_type: XLA
         downcast_bf16: 'no'
         machine_rank: 0
         main_training_function: main