Skip to content

Commit d128ee6

Browse files
authored
[E2E] Add accelerate dependency for some huggingface models (#5359)
Signed-off-by: Anatoly Myachev <[email protected]>
1 parent 3b29f6d commit d128ee6

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

.github/workflows/e2e-reusable.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,7 @@ jobs:
144144
run: |
145145
cd pytorch
146146
pip install -r .ci/docker/ci_commit_pins/huggingface-requirements.txt
147+
pip install accelerate
147148
148149
- name: Install torchvision package
149150
if: ${{ inputs.suite == 'timm_models' || inputs.suite == 'torchbench' }}

0 commit comments

Comments
 (0)