From 7ef0084b0d2e3b91fe1fa7cd5e396d47aa613797 Mon Sep 17 00:00:00 2001 From: Lianmin Zheng Date: Fri, 8 Nov 2024 01:21:29 -0800 Subject: [PATCH] Add sentence_transformers to CI dependency (#1958) --- scripts/ci_install_dependency.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/ci_install_dependency.sh b/scripts/ci_install_dependency.sh index a13cfd214..a219e02e2 100644 --- a/scripts/ci_install_dependency.sh +++ b/scripts/ci_install_dependency.sh @@ -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