ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[misc] fix ds config (#7205)

Browse Source 
Former-commit-id: b478fa1d9de1858075769f86f57126fde92db813
This commit is contained in:
hoshi-hiyouga
2025-03-07 15:21:28 +08:00
committed by GitHub
parent 151ef48b40
commit 82a2bac866
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/accelerate/fsdp_config.yaml
View File

@@ -14,7 +14,7 @@ fsdp_config:
fsdp_use_orig_params: true fsdp_use_orig_params: true
machine_rank: 0 machine_rank: 0
main_training_function: main main_training_function: main
mixed_precision: fp16 # or bf16 mixed_precision: bf16 # or fp16
num_machines: 1 # the number of nodes num_machines: 1 # the number of nodes
num_processes: 2 # the number of GPUs in all nodes num_processes: 2 # the number of GPUs in all nodes
rdzv_backend: static rdzv_backend: static