Skip to content

Commit 8c7f63c

Browse files
authored
fix: unpack cce imported incorrectly (#3212) [skip ci]
1 parent cd856b4 commit 8c7f63c

File tree

4 files changed

+4
-4
lines changed

4 files changed

+4
-4
lines changed

examples/colab-notebooks/colab-axolotl-example.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@
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
{

scripts/cutcrossentropy_install.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,5 +29,5 @@
2929

3030
print(
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
)

src/axolotl/integrations/cut_cross_entropy/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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

src/axolotl/integrations/cut_cross_entropy/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@
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

0 commit comments

Comments
 (0)