423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-16 11:50:35 +08:00
Code Issues 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 ee9bd3f2d7
commit 678c65fc28
6 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/deepspeed/ds_z0_config.json
View File

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

2
examples/deepspeed/ds_z2_config.json
View File

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

2
examples/deepspeed/ds_z2_offload_config.json
View File

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

2
examples/deepspeed/ds_z3_config.json
View File

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

2
examples/deepspeed/ds_z3_offload_config.json
View File

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