Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/torchani.git" did not exist on "3018a82ba7650302f9010d5b41d0b0098cc4d13b"
Unverified Commit f141298a authored by Lianmin Zheng's avatar Lianmin Zheng Committed by GitHub
Browse files

Update ci_install_dependency.sh to use accelerate 1.4.0 (#4392) 


Co-authored-by: default avatarwangyu <wangyu.steph@bytedance.com>
Co-authored-by: default avatarwangyu <yuwangauto@foxmail.com>
parent 4fea040c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
scripts/ci_install_dependency.sh
View file @ f141298a
...@@ -20,7 +20,7 @@ pip install flashinfer_python==0.2.3 --find-links ${FLASHINFER_REPO} --force-rei ...@@ -20,7 +20,7 @@ pip install flashinfer_python==0.2.3 --find-links ${FLASHINFER_REPO} --force-rei
pip install torch_memory_saver --force-reinstall pip install torch_memory_saver --force-reinstall
pip install transformers==4.45.2 sentence_transformers accelerate peft pandas datasets pip install transformers==4.45.2 sentence_transformers accelerate==1.4.0 peft pandas datasets
# For compling xgrammar kernels # For compling xgrammar kernels
pip install cuda-python nvidia-cuda-nvrtc-cu12 pip install cuda-python nvidia-cuda-nvrtc-cu12
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment