Add sentence_transformers to CI dependency (#1958)

This commit is contained in:
Lianmin Zheng
2024-11-08 01:21:29 -08:00
committed by GitHub
parent f9a377f650
commit 7ef0084b0d

View File

@@ -4,5 +4,5 @@ Install the dependency in CI.
pip install --upgrade pip
pip install -e "python[all]"
pip install transformers==4.45.2
pip install transformers==4.45.2 sentence_transformers
pip install flashinfer -i https://flashinfer.ai/whl/cu121/torch2.4/ --force-reinstall