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

Use Latest Cut Cross Entropy (#2392)

Browse Source 
* Update __init__.py

* Update README.md

* Update cutcrossentropy_install.py

* add test
This commit is contained in:
xzuyn
2025-03-10 05:26:40 -04:00
committed by GitHub
parent 46a045e528
commit 60a11a6410
4 changed files with 48 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/cutcrossentropy_install.py
View File

@@ -24,5 +24,5 @@ if cce_spec:
print(
UNINSTALL_PREFIX
+ 'pip install "cut-cross-entropy @ git+https://github.com/apple/ml-cross-entropy.git@9c297c905f55b73594b5d650722d1e78183b77bd"'
+ 'pip install "cut-cross-entropy[transformers] @ git+https://github.com/apple/ml-cross-entropy.git@24fbe4b5dab9a6c250a014573613c1890190536c"'
)