Skip to content
This repository was archived by the owner on Jun 22, 2025. It is now read-only.

Commit 1c1849f

Browse files
authored
Use model cache for llms docker compose test. (opea-project#1463)
Signed-off-by: ZePan110 <ze.pan@intel.com>
1 parent b6f73e5 commit 1c1849f

12 files changed

Lines changed: 12 additions & 7 deletions

tests/llms/test_llms_doc-summarization_tgi.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ export REGISTRY=${IMAGE_REPO}
99
export TAG="comps"
1010
echo "REGISTRY=IMAGE_REPO=${IMAGE_REPO}"
1111
echo "TAG=${TAG}"
12+
export DATA_PATH=${model_cache}
1213

1314
WORKPATH=$(dirname "$PWD")
1415
host_ip=$(hostname -I | awk '{print $1}')

tests/llms/test_llms_doc-summarization_tgi_on_intel_hpu.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ function start_service() {
3636
export MAX_INPUT_TOKENS=2048
3737
export MAX_TOTAL_TOKENS=4096
3838
export LOGFLAG=True
39-
export DATA_PATH="/data2/hf_model"
39+
export DATA_PATH=${model_cache:-./data}
4040

4141
cd $WORKPATH/comps/llms/deployment/docker_compose
4242
docker compose -f compose_doc-summarization.yaml up ${service_name} -d > ${LOG_PATH}/start_services_with_compose.log

tests/llms/test_llms_doc-summarization_vllm.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ export REGISTRY=${IMAGE_REPO}
99
export TAG="comps"
1010
echo "REGISTRY=IMAGE_REPO=${IMAGE_REPO}"
1111
echo "TAG=${TAG}"
12+
export DATA_PATH=${model_cache}
1213

1314
WORKPATH=$(dirname "$PWD")
1415
host_ip=$(hostname -I | awk '{print $1}')

tests/llms/test_llms_doc-summarization_vllm_on_intel_hpu.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ function start_service() {
5151
export MAX_TOTAL_TOKENS=4096
5252
export VLLM_SKIP_WARMUP=true
5353
export LOGFLAG=True
54-
export DATA_PATH="/data2/hf_model"
54+
export DATA_PATH=${model_cache:-./data}
5555

5656
cd $WORKPATH/comps/llms/deployment/docker_compose
5757
docker compose -f compose_doc-summarization.yaml up ${service_name} -d > ${LOG_PATH}/start_services_with_compose.log

tests/llms/test_llms_faq-generation_tgi.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ export REGISTRY=${IMAGE_REPO}
99
export TAG="comps"
1010
echo "REGISTRY=IMAGE_REPO=${IMAGE_REPO}"
1111
echo "TAG=${TAG}"
12+
export DATA_PATH=${model_cache}
1213

1314
WORKPATH=$(dirname "$PWD")
1415
host_ip=$(hostname -I | awk '{print $1}')

tests/llms/test_llms_faq-generation_tgi_on_intel_hpu.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ function start_service() {
3434
export LLM_ENDPOINT="http://${host_ip}:${LLM_ENDPOINT_PORT}"
3535
export LLM_MODEL_ID="Intel/neural-chat-7b-v3-3"
3636
export LOGFLAG=True
37-
export DATA_PATH="/data2/hf_model"
37+
export DATA_PATH=${model_cache:-./data}
3838

3939
cd $WORKPATH/comps/llms/deployment/docker_compose
4040
docker compose -f compose_faq-generation.yaml up ${service_name} -d > ${LOG_PATH}/start_services_with_compose.log

tests/llms/test_llms_faq-generation_vllm.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ export REGISTRY=${IMAGE_REPO}
99
export TAG="comps"
1010
echo "REGISTRY=IMAGE_REPO=${IMAGE_REPO}"
1111
echo "TAG=${TAG}"
12+
export DATA_PATH=${model_cache}
1213

1314
WORKPATH=$(dirname "$PWD")
1415
host_ip=$(hostname -I | awk '{print $1}')

tests/llms/test_llms_faq-generation_vllm_on_intel_hpu.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ function start_service() {
5050
export LLM_MODEL_ID="Intel/neural-chat-7b-v3-3"
5151
export VLLM_SKIP_WARMUP=true
5252
export LOGFLAG=True
53-
export DATA_PATH="/data2/hf_model"
53+
export DATA_PATH=${model_cache:-./data}
5454

5555
cd $WORKPATH/comps/llms/deployment/docker_compose
5656
docker compose -f compose_faq-generation.yaml up ${service_name} -d > ${LOG_PATH}/start_services_with_compose.log

tests/llms/test_llms_text-generation_native_on_intel_hpu.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ function start_service() {
3131
export host_ip=${host_ip}
3232
export LLM_MODEL_ID="Intel/neural-chat-7b-v3-3"
3333
export LOGFLAG=True
34-
export DATA_PATH="/data2/hf_model"
34+
export DATA_PATH=${model_cache:-./data}
3535

3636
cd $WORKPATH/comps/llms/deployment/docker_compose
3737
docker compose -f compose_text-generation.yaml up ${service_name} -d > ${LOG_PATH}/start_services_with_compose.log

tests/llms/test_llms_text-generation_service_tgi.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ export REGISTRY=${IMAGE_REPO}
99
export TAG="comps"
1010
echo "REGISTRY=IMAGE_REPO=${IMAGE_REPO}"
1111
echo "TAG=${TAG}"
12+
export DATA_PATH=${model_cache}
1213

1314
WORKPATH=$(dirname "$PWD")
1415
host_ip=$(hostname -I | awk '{print $1}')

0 commit comments

Comments
 (0)