mirror of https://github.com/vllm-project/vllm
[ci] Add AWS creds for AMD (#13572)
This commit is contained in:
parent
512368e34a
commit
88f6ba3281
|
@ -121,6 +121,8 @@ if [[ $commands == *"--shard-id="* ]]; then
|
|||
--rm \
|
||||
-e HIP_VISIBLE_DEVICES="${GPU}" \
|
||||
-e HF_TOKEN \
|
||||
-e AWS_ACCESS_KEY_ID \
|
||||
-e AWS_SECRET_ACCESS_KEY \
|
||||
-v "${HF_CACHE}:${HF_MOUNT}" \
|
||||
-e "HF_HOME=${HF_MOUNT}" \
|
||||
--name "${container_name}_${GPU}" \
|
||||
|
@ -148,6 +150,8 @@ else
|
|||
--rm \
|
||||
-e HIP_VISIBLE_DEVICES=0 \
|
||||
-e HF_TOKEN \
|
||||
-e AWS_ACCESS_KEY_ID \
|
||||
-e AWS_SECRET_ACCESS_KEY \
|
||||
-v "${HF_CACHE}:${HF_MOUNT}" \
|
||||
-e "HF_HOME=${HF_MOUNT}" \
|
||||
--name "${container_name}" \
|
||||
|
|
|
@ -10,3 +10,5 @@ ray >= 2.10.0
|
|||
peft
|
||||
pytest-asyncio
|
||||
tensorizer>=2.9.0
|
||||
runai-model-streamer==0.11.0
|
||||
runai-model-streamer-s3==0.11.0
|
||||
|
|
Loading…
Reference in New Issue