Skip to content

GitLab

"vllm/vscode:/vscode.git/clone" did not exist on "44d02f54db522fc489d3590d5de9461a52621460"
Unverified Commit 88f6ba32 authored by Kevin H. Luu's avatar Kevin H. Luu Committed by GitHub
Browse files

[ci] Add AWS creds for AMD (#13572)

parent 512368e3
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 0 deletions
.buildkite/run-amd-test.sh
View file @ 88f6ba32
...@@ -121,6 +121,8 @@ if [[ $commands == *"--shard-id="* ]]; then ...@@ -121,6 +121,8 @@ if [[ $commands == *"--shard-id="* ]]; then
--rm \ --rm \
-e HIP_VISIBLE_DEVICES="${GPU}" \ -e HIP_VISIBLE_DEVICES="${GPU}" \
-e HF_TOKEN \ -e HF_TOKEN \
-e AWS_ACCESS_KEY_ID \
-e AWS_SECRET_ACCESS_KEY \
-v "${HF_CACHE}:${HF_MOUNT}" \ -v "${HF_CACHE}:${HF_MOUNT}" \
-e "HF_HOME=${HF_MOUNT}" \ -e "HF_HOME=${HF_MOUNT}" \
--name "${container_name}_${GPU}" \ --name "${container_name}_${GPU}" \
...@@ -148,6 +150,8 @@ else ...@@ -148,6 +150,8 @@ else
--rm \ --rm \
-e HIP_VISIBLE_DEVICES=0 \ -e HIP_VISIBLE_DEVICES=0 \
-e HF_TOKEN \ -e HF_TOKEN \
-e AWS_ACCESS_KEY_ID \
-e AWS_SECRET_ACCESS_KEY \
-v "${HF_CACHE}:${HF_MOUNT}" \ -v "${HF_CACHE}:${HF_MOUNT}" \
-e "HF_HOME=${HF_MOUNT}" \ -e "HF_HOME=${HF_MOUNT}" \
--name "${container_name}" \ --name "${container_name}" \
......
requirements-rocm.txt
View file @ 88f6ba32
...@@ -10,3 +10,5 @@ ray >= 2.10.0 ...@@ -10,3 +10,5 @@ ray >= 2.10.0
peft peft
pytest-asyncio pytest-asyncio
tensorizer>=2.9.0 tensorizer>=2.9.0
runai-model-streamer==0.11.0
runai-model-streamer-s3==0.11.0
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment