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

fix?

Browse Source
This commit is contained in:
Dan Saunders
2025-10-01 16:18:40 -04:00
parent 00e0238501
commit ccd2f12335
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/cutcrossentropy_install.py
View File

@@ -27,13 +27,13 @@ UNINSTALL_PREFIX = ""
if cce_spec: if cce_spec:
if not importlib.util.find_spec("cut_cross_entropy.transformers"): if not importlib.util.find_spec("cut_cross_entropy.transformers"):
if USE_UV: if USE_UV:
UNINSTALL_PREFIX = "uv pip uninstall --system --yes cut-cross-entropy && " UNINSTALL_PREFIX = "uv pip uninstall --yes cut-cross-entropy && "
else: else:
UNINSTALL_PREFIX = "pip uninstall -y cut-cross-entropy && " UNINSTALL_PREFIX = "pip uninstall -y cut-cross-entropy && "
installer = "uv pip --system" if USE_UV else "pip" installer = "uv pip install --system" if USE_UV else "pip install"
command = ( command = (
f"{installer} install " f"{installer} "
'"cut-cross-entropy[transformers] ' '"cut-cross-entropy[transformers] '
'@ git+https://github.com/axolotl-ai-cloud/ml-cross-entropy.git@147ea28"' '@ git+https://github.com/axolotl-ai-cloud/ml-cross-entropy.git@147ea28"'
) )