File tree Expand file tree Collapse file tree 4 files changed +4
-4
lines changed
src/axolotl/integrations/cut_cross_entropy Expand file tree Collapse file tree 4 files changed +4
-4
lines changed Original file line number Diff line number Diff line change 4040 " %%capture\n " ,
4141 " # This step can take ~5-10 minutes to install dependencies\n " ,
4242 " !pip install --no-build-isolation axolotl[flash-attn]>=0.9.1\n " ,
43- " !pip install \" cut-cross-entropy[transformers] @ git+https://github.com/axolotl-ai-cloud/ml-cross-entropy.git@49f3308 \" "
43+ " !pip install \" cut-cross-entropy[transformers] @ git+https://github.com/axolotl-ai-cloud/ml-cross-entropy.git@8a1a0ec \" "
4444 ]
4545 },
4646 {
Original file line number Diff line number Diff line change 2929
3030print (
3131 UNINSTALL_PREFIX
32- + f'{ UV_PREFIX } pip install "cut-cross-entropy[transformers] @ git+https://github.com/axolotl-ai-cloud/ml-cross-entropy.git@49f3308 "'
32+ + f'{ UV_PREFIX } pip install "cut-cross-entropy[transformers] @ git+https://github.com/axolotl-ai-cloud/ml-cross-entropy.git@8a1a0ec "'
3333)
Original file line number Diff line number Diff line change @@ -19,7 +19,7 @@ python scripts/cutcrossentropy_install.py | sh
1919
2020- If you are installing from pip
2121``` bash
22- pip3 uninstall -y cut-cross-entropy && pip3 install " cut-cross-entropy[transformers] @ git+https://github.com/axolotl-ai-cloud/ml-cross-entropy.git@49f3308 "
22+ pip3 uninstall -y cut-cross-entropy && pip3 install " cut-cross-entropy[transformers] @ git+https://github.com/axolotl-ai-cloud/ml-cross-entropy.git@8a1a0ec "
2323```
2424
2525## Usage
Original file line number Diff line number Diff line change 3535
3636_CCE_INSTALL_MESSAGE = (
3737 "Please install Axolotl's fork of cut_cross_entropy with transformers support using "
38- '`pip install "cut-cross-entropy[transformers] @ git+https://github.com/axolotl-ai-cloud/ml-cross-entropy.git@49f3308 "`'
38+ '`pip install "cut-cross-entropy[transformers] @ git+https://github.com/axolotl-ai-cloud/ml-cross-entropy.git@8a1a0ec "`'
3939)
4040
4141
You can’t perform that action at this time.
0 commit comments