423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-01 03:02:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

[misc] fix ds config (#7205)

Browse Source 
Former-commit-id: db113f690eb818894e3bc2bfc69cdfd27045d808
This commit is contained in:
hoshi-hiyouga 2025-03-07 15:21:28 +08:00 committed by GitHub
parent 33b4c33279
commit 98ea0e8109
6 changed files with 6 additions and 6 deletions
Show Stats 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

2
examples/deepspeed/ds_z0_config.json
View File

@ -19,7 +19,7 @@
"stage": 0, "stage": 0,
"allgather_partitions": true, "allgather_partitions": true,
"allgather_bucket_size": 5e8, "allgather_bucket_size": 5e8,
"overlap_comm": true, "overlap_comm": false,
"reduce_scatter": true, "reduce_scatter": true,
"reduce_bucket_size": 5e8, "reduce_bucket_size": 5e8,
"contiguous_gradients": true, "contiguous_gradients": true,

2
examples/deepspeed/ds_z2_config.json
View File

@ -19,7 +19,7 @@
"stage": 2, "stage": 2,
"allgather_partitions": true, "allgather_partitions": true,
"allgather_bucket_size": 5e8, "allgather_bucket_size": 5e8,
"overlap_comm": true, "overlap_comm": false,
"reduce_scatter": true, "reduce_scatter": true,
"reduce_bucket_size": 5e8, "reduce_bucket_size": 5e8,
"contiguous_gradients": true, "contiguous_gradients": true,

2
examples/deepspeed/ds_z2_offload_config.json
View File

@ -23,7 +23,7 @@
}, },
"allgather_partitions": true, "allgather_partitions": true,
"allgather_bucket_size": 5e8, "allgather_bucket_size": 5e8,
"overlap_comm": true, "overlap_comm": false,
"reduce_scatter": true, "reduce_scatter": true,
"reduce_bucket_size": 5e8, "reduce_bucket_size": 5e8,
"contiguous_gradients": true, "contiguous_gradients": true,

2
examples/deepspeed/ds_z3_config.json
View File

@ -17,7 +17,7 @@
}, },
"zero_optimization": { "zero_optimization": {
"stage": 3, "stage": 3,
"overlap_comm": true, "overlap_comm": false,
"contiguous_gradients": true, "contiguous_gradients": true,
"sub_group_size": 1e9, "sub_group_size": 1e9,
"reduce_bucket_size": "auto", "reduce_bucket_size": "auto",

2
examples/deepspeed/ds_z3_offload_config.json
View File

@ -25,7 +25,7 @@
"device": "cpu", "device": "cpu",
"pin_memory": true "pin_memory": true
}, },
"overlap_comm": true, "overlap_comm": false,
"contiguous_gradients": true, "contiguous_gradients": true,
"sub_group_size": 1e9, "sub_group_size": 1e9,
"reduce_bucket_size": "auto", "reduce_bucket_size": "auto",