From 82a2bac866c396d21f3cf8b33b3238b8631f3825 Mon Sep 17 00:00:00 2001 From: hoshi-hiyouga Date: Fri, 7 Mar 2025 15:21:28 +0800 Subject: [PATCH] [misc] fix ds config (#7205) Former-commit-id: b478fa1d9de1858075769f86f57126fde92db813 --- examples/accelerate/fsdp_config.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/accelerate/fsdp_config.yaml b/examples/accelerate/fsdp_config.yaml index cd65e074..6fb09a95 100644 --- a/examples/accelerate/fsdp_config.yaml +++ b/examples/accelerate/fsdp_config.yaml @@ -14,7 +14,7 @@ fsdp_config: fsdp_use_orig_params: true machine_rank: 0 main_training_function: main -mixed_precision: fp16 # or bf16 +mixed_precision: bf16 # or fp16 num_machines: 1 # the number of nodes num_processes: 2 # the number of GPUs in all nodes rdzv_backend: static