Update evaluate.yml

zhulinJulia24 · Feb 8, 2025 · 11fab06 · 11fab06
1 parent 52c74ef
commit 11fab06
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/evaluate.yml b/.github/workflows/evaluate.yml
@@ -134,7 +134,7 @@ jobs:
           # manually install flash attn
           # the install packeage from. https://github.com/Dao-AILab/flash-attention/releases
           python3 -m pip install /root/packages/flash_attn-*.whl
-          python3 -m pip install sentence_transformers --no-deps
+          python3 -m pip install sentence_transformers==2.2.2 --no-deps
           python3 -m pip install -r /root/models/offline_pkg/requirements.txt
       - name: Install lmdeploy
         if: ${{github.event_name == 'schedule' || !inputs.offline_mode}}