Update ci_install_dependency.sh to use accelerate 1.4.0 (#4392)
Co-authored-by: wangyu <wangyu.steph@bytedance.com> Co-authored-by: wangyu <yuwangauto@foxmail.com>
This commit is contained in:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user