This commit is contained in:
Dan Saunders
2025-10-02 12:07:58 -04:00
parent 5771a65b88
commit 98730868e7
2 changed files with 14 additions and 9 deletions

View File

@@ -3,6 +3,7 @@
from __future__ import annotations
import sys
from shlex import quote
try:
import torch
@@ -15,8 +16,9 @@ if V(torch.__version__.split("+")[0]) < V("2.6.0"):
print("")
sys.exit(0)
python_exe = quote(sys.executable)
print(
"python -m pip install "
f"{python_exe} -m pip install "
'"cut-cross-entropy[transformers] '
'@ git+https://github.com/axolotl-ai-cloud/ml-cross-entropy.git@147ea28"'
)