github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-13 19:46:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

remote comments

Browse Source
This commit is contained in:
genghaozhe
2024-05-20 05:15:51 +00:00
parent 7416e4943b
commit df63db7e63
6 changed files with 148 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_zero/test_gemini/test_optim.py
View File

@@ -26,7 +26,7 @@ PLACEMENT_CONFIGS = [
"offload_optim_frac": 1.0,
"offload_param_frac": 1.0,
}, # zero3-offload-all
{"placement_policy": "auto"},
# {"placement_policy": "auto"},
]
# this model is large enough to slice to chunks