ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

fix transformer version (#10788)

Browse source 
* fix transformer version

* uninstall sentence transformer

* uninstall

* uninstall
This commit is contained in:
Zhicun 2024-04-18 17:37:21 +08:00 committed by GitHub
parent 0e8aac19e3
commit 88463cbf47
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 13 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
.github/workflows/llm_unit_tests.yml vendored
View file

@ -223,9 +223,14 @@ jobs:
shell: bash
run: |
pip install llama-index-readers-file llama-index-vector-stores-postgres llama-index-embeddings-huggingface
pip install transformers==4.31.0
pip install transformers==4.36.0
pip install "pydantic>=2.0.0"
bash python/llm/test/run-llm-llamaindex-tests.sh
- name: Run sentence-transformers uninstallation
if: ${{ always() }}
shell: bash
run: |
pip uninstall sentence-transformers -y || true
llm-unit-test-on-arc:
needs: [setup-python-version, llm-cpp-build]
strategy:
@ -399,5 +404,11 @@ jobs:
pip install --pre --upgrade ipex-llm[xpu_2.0] --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/
source /home/arda/intel/oneapi/setvars.sh
fi
pip install transformers==4.36.0
pip install "pydantic>=2.0.0"
bash python/llm/test/run-llm-llamaindex-tests-gpu.sh
bash python/llm/test/run-llm-llamaindex-tests-gpu.sh
- name: Run sentence-transformers uninstallation
if: ${{ always() }}
shell: bash
run: |
pip uninstall sentence-transformers -y || true