mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-05-12 07:40:13 +08:00
[CI][TPU] Fix docker clean up (#22271)
Signed-off-by: Siyuan Liu <lsiyuan@google.com>
This commit is contained in:
parent
59a0b8554b
commit
4b29d2784b
@ -4,8 +4,7 @@ set -xu
|
|||||||
|
|
||||||
|
|
||||||
remove_docker_container() {
|
remove_docker_container() {
|
||||||
docker rm -f tpu-test || true;
|
docker rm -f tpu-test || true;
|
||||||
docker rm -f vllm-tpu || true;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
trap remove_docker_container EXIT
|
trap remove_docker_container EXIT
|
||||||
|
|||||||
@ -5,7 +5,6 @@ set -xu
|
|||||||
|
|
||||||
remove_docker_container() {
|
remove_docker_container() {
|
||||||
docker rm -f tpu-test || true;
|
docker rm -f tpu-test || true;
|
||||||
docker rm -f vllm-tpu || true;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
trap remove_docker_container EXIT
|
trap remove_docker_container EXIT
|
||||||
|
|||||||
@ -1,6 +1,6 @@
|
|||||||
# Environment config
|
# Environment config
|
||||||
TEST_NAME=llama8b
|
TEST_NAME=llama8b
|
||||||
CONTAINER_NAME=vllm-tpu
|
CONTAINER_NAME=tpu-test
|
||||||
|
|
||||||
# vllm config
|
# vllm config
|
||||||
MODEL=meta-llama/Llama-3.1-8B-Instruct
|
MODEL=meta-llama/Llama-3.1-8B-Instruct
|
||||||
|
|||||||
@ -12,8 +12,6 @@ source /etc/environment
|
|||||||
source $ENV_FILE
|
source $ENV_FILE
|
||||||
|
|
||||||
remove_docker_container() {
|
remove_docker_container() {
|
||||||
docker rm -f tpu-test || true;
|
|
||||||
docker rm -f vllm-tpu || true;
|
|
||||||
docker rm -f $CONTAINER_NAME || true;
|
docker rm -f $CONTAINER_NAME || true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@ -1,6 +1,6 @@
|
|||||||
# Environment config
|
# Environment config
|
||||||
TEST_NAME=llama8bw8a8
|
TEST_NAME=llama8bw8a8
|
||||||
CONTAINER_NAME=vllm-tpu
|
CONTAINER_NAME=tpu-test
|
||||||
|
|
||||||
# vllm config
|
# vllm config
|
||||||
MODEL=RedHatAI/Meta-Llama-3.1-8B-Instruct-quantized.w8a8
|
MODEL=RedHatAI/Meta-Llama-3.1-8B-Instruct-quantized.w8a8
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user