Skip to content

Commit 3ceb8ff

Browse files
committed
direct install dependency
Signed-off-by: Maanu Grover <[email protected]>
1 parent ce28a1f commit 3ceb8ff

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

.github/workflows/cache-hf-model.yml

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,6 @@ jobs:
2929
name: Download Model
3030
runs-on: self-hosted-nemo
3131
environment: nemo-ci
32-
container:
33-
image: nvcr.io/nvidia/pytorch:25.05-py3
3432
env:
3533
HF_HUB_OFFLINE: 0
3634
HF_HOME: /mnt/datadrive/TestData/HF_HOME
@@ -42,7 +40,7 @@ jobs:
4240

4341
- name: Install dependencies
4442
run: |
45-
pip install transformers
43+
pip install transformers torch
4644
4745
- name: Download model
4846
env:

0 commit comments

Comments
 (0)