[Feat] upgrade pytorch2.6 (#5417)

This commit is contained in:
lukec
2025-04-21 07:06:34 +08:00
committed by GitHub
parent 475e2e378a
commit 417b44eba8
7 changed files with 8 additions and 8 deletions

View File

@@ -23,7 +23,7 @@ pip install -e "python[all]"
# Install additional dependencies
pip install torch_memory_saver
pip install transformers==4.51.0 sentence_transformers accelerate==1.4.0 peft pandas datasets timm torchaudio
pip install transformers==4.51.0 sentence_transformers accelerate peft pandas datasets timm torchaudio
# For compling xgrammar kernels
pip install cuda-python nvidia-cuda-nvrtc-cu12