Skip to content

Commit d21bb9a

Browse files
authored
pin transformers for llama testing and mpmath for shark testing (#477)
Pin transformers to version 4.37.1 and mpmath to 1.3 to fix the testing failures with the new version until we get it fixed. Please see #476 for details on reason for testing failure with updated transformers.
1 parent 04c1a01 commit d21bb9a

File tree

4 files changed

+5
-3
lines changed

4 files changed

+5
-3
lines changed

.github/workflows/test_shark.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,10 +42,11 @@ jobs:
4242
cd $GITHUB_WORKSPACE/SHARK
4343
python${{ matrix.version }} -m venv shark.venv
4444
source shark.venv/bin/activate
45-
sed -i 's/SHARK-Turbine.git@main/SHARK-Turbine.git@${{github.sha}}/g' requirements.txt
4645
sed -i 's/SHARK-Turbine#/SHARK-Turbine.git@${{github.sha}}#/g' requirements.txt
4746
pip install -r requirements.txt --no-cache-dir
4847
pip install -e .
4948
pip uninstall -y torch
5049
pip install torch==2.1.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
50+
pip uninstall -y mpmath
51+
pip install mpmath==1.3.0
5152
python apps/shark_studio/tests/api_test.py

core/pytorch-cpu-requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,3 @@
11
--pre
22
torch==2.1.0
3+
mpmath==1.3.0

models/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
protobuf
22
sentencepiece
33
shark_turbine
4-
transformers
4+
transformers==4.37.1
55
accelerate
66
diffusers==0.24.0
77
brevitas @ git+https://github.com/Xilinx/brevitas.git@6695e8df7f6a2c7715b9ed69c4b78157376bb60b

models/setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ def load_version_info():
5858
"brevitas @ git+https://github.com/Xilinx/brevitas.git@6695e8df7f6a2c7715b9ed69c4b78157376bb60b",
5959
"protobuf",
6060
"sentencepiece",
61-
"transformers",
61+
"transformers==4.37.1",
6262
"accelerate",
6363
"diffusers==0.24.0",
6464
],

0 commit comments

Comments
 (0)