Skip to content

Commit 88f6ba3

Browse files
authored
[ci] Add AWS creds for AMD (#13572)
1 parent 512368e commit 88f6ba3

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

.buildkite/run-amd-test.sh

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -121,6 +121,8 @@ if [[ $commands == *"--shard-id="* ]]; then
121121
--rm \
122122
-e HIP_VISIBLE_DEVICES="${GPU}" \
123123
-e HF_TOKEN \
124+
-e AWS_ACCESS_KEY_ID \
125+
-e AWS_SECRET_ACCESS_KEY \
124126
-v "${HF_CACHE}:${HF_MOUNT}" \
125127
-e "HF_HOME=${HF_MOUNT}" \
126128
--name "${container_name}_${GPU}" \
@@ -148,6 +150,8 @@ else
148150
--rm \
149151
-e HIP_VISIBLE_DEVICES=0 \
150152
-e HF_TOKEN \
153+
-e AWS_ACCESS_KEY_ID \
154+
-e AWS_SECRET_ACCESS_KEY \
151155
-v "${HF_CACHE}:${HF_MOUNT}" \
152156
-e "HF_HOME=${HF_MOUNT}" \
153157
--name "${container_name}" \

requirements-rocm.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,3 +10,5 @@ ray >= 2.10.0
1010
peft
1111
pytest-asyncio
1212
tensorizer>=2.9.0
13+
runai-model-streamer==0.11.0
14+
runai-model-streamer-s3==0.11.0

0 commit comments

Comments
 (0)