tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Mixtral fixes 20240124 (#1192) [skip ci]

Browse Source 
* mixtral nccl fixes

* make sure to patch for z3
This commit is contained in:
Wing Lian
2024-01-24 14:59:57 -05:00
committed by GitHub
parent af0243021c
commit 54d2ac155b
14 changed files with 71 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/mistral/README.md
View File

@@ -8,5 +8,5 @@ accelerate launch -m axolotl.cli.train examples/mistral/config.yml
If you run into CUDA OOM, use deepspeed with config zero2.json:
```shell
accelerate launch -m axolotl.cli.train examples/mistral/config.yml --deepspeed deepspeed/zero2.json
accelerate launch -m axolotl.cli.train examples/mistral/config.yml --deepspeed deepspeed_configs/zero2.json
```