Fix issue (opea-project#483)

* Standardize image names and labels. Signed-off-by: zepan <[email protected]> * Remove test cases that are no longer in use. Signed-off-by: zepan <[email protected]> --------- Signed-off-by: zepan <[email protected]>
predictionguard · Sep 3, 2024 · ea20181 · ea20181
1 parent 6ae0dd4
commit ea20181
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 99 deletions.
diff --git a/tests/test_llms_text-generation_ray_serve.sh b/tests/test_llms_text-generation_ray_serve.sh
diff --git a/tests/test_llms_text-generation_vllm-ray.sh b/tests/test_llms_text-generation_vllm-ray.sh
@@ -34,7 +34,7 @@ function start_service() {
         --ipc=host \
         -e HUGGINGFACEHUB_API_TOKEN=$HUGGINGFACEHUB_API_TOKEN \
         -p $port_number:8000 \
-        opea/vllm_ray:habana \
+        opea/vllm_ray-habana:comps \
         /bin/bash -c "ray start --head && python vllm_ray_openai.py --port_number 8000 --model_id_or_path $LLM_MODEL --tensor_parallel_size 2 --enforce_eager False"
 
     export vLLM_RAY_ENDPOINT="http://${ip_address}:${port_number}"