Skip to content

Commit 78169ef

Browse files
committed
Define version of safetensor for Huggingface workflow
Signed-off-by: Beat Buesser <[email protected]>
1 parent 14be728 commit 78169ef

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

.github/workflows/ci-huggingface.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,13 +51,14 @@ jobs:
5151
sudo apt-get update
5252
sudo apt-get -y -q install ffmpeg libavcodec-extra
5353
python -m pip install --upgrade pip setuptools wheel
54-
pip install -q -r <(sed '/^tensorflow/d;/^keras/d;/^torch/d;/^torchvision/d;/^torchaudio/d;/^transformers/d' requirements_test.txt)
54+
pip install -q -r <(sed '/^tensorflow/d;/^keras/d;/^torch/d;/^torchvision/d;/^torchaudio/d;/^transformers/d;/^safetensors/d' requirements_test.txt)
5555
pip install tensorflow==2.18.1
5656
pip install keras==3.10.0
5757
pip install torch==${{ matrix.torch }} --index-url https://download.pytorch.org/whl/cpu
5858
pip install torchvision==${{ matrix.torchvision }} --index-url https://download.pytorch.org/whl/cpu
5959
pip install torchaudio==${{ matrix.torchaudio }} --index-url https://download.pytorch.org/whl/cpu
6060
pip install transformers==${{ matrix.transformers }}
61+
pip install safetensors==0.5.3
6162
pip list
6263
6364
- name: Cache CIFAR-10 dataset

0 commit comments

Comments
 (0)