Skip to content

Commit 4252121

Browse files
authored
Use model cache for dataprep docker compose test. (#1450)
Use model cache for dataprep docker compose test. Fix web_retrievers typo Signed-off-by: ZePan110 <ze.pan@intel.com>
1 parent 0672112 commit 4252121

File tree

9 files changed

+11
-6
lines changed

9 files changed

+11
-6
lines changed

.github/workflows/manual-comps-test.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,4 +69,5 @@ jobs:
6969
mode: ${{ inputs.mode }}
7070
test_compose: ${{ inputs.test_compose }}
7171
test_helmchart: ${{ inputs.test_helmchart }}
72+
use_model_cache: ${{ inputs.use_model_cache }}
7273
secrets: inherit

comps/third_parties/tei/deployment/docker_compose/compose.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ services:
99
ports:
1010
- "${TEI_EMBEDDER_PORT:-12000}:80"
1111
volumes:
12-
- "${DATA_PATH-./data}:/data"
12+
- "${DATA_PATH:-./data}:/data"
1313
shm_size: 1g
1414
environment:
1515
no_proxy: ${no_proxy}
@@ -30,7 +30,7 @@ services:
3030
ports:
3131
- "${TEI_RERANKING_PORT:-12003}:80"
3232
volumes:
33-
- "${DATA_PATH-./data}:/data"
33+
- "${DATA_PATH:-./data}:/data"
3434
shm_size: 1g
3535
environment:
3636
no_proxy: ${no_proxy}
@@ -51,7 +51,7 @@ services:
5151
ports:
5252
- ${TEI_RERANKING_PORT:-12004}:80
5353
volumes:
54-
- "${DATA_PATH-./data}:/data"
54+
- "${DATA_PATH:-./data}:/data"
5555
shm_size: 1g
5656
environment:
5757
no_proxy: ${no_proxy}

comps/web_retrievers/deployment/docker_compose/compose.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ services:
2727
ports:
2828
- ${TEI_PORT:-6060}:80
2929
volumes:
30-
- "${DATA_PATH-./data}:/data"
30+
- "${DATA_PATH:-./data}:/data"
3131
shm_size: 1g
3232
environment:
3333
no_proxy: ${no_proxy}

tests/dataprep/test_dataprep_milvus.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ ip_address=$(hostname -I | awk '{print $1}')
1010
DATAPREP_PORT=11101
1111
service_name="dataprep-milvus tei-embedding-serving etcd minio standalone"
1212
export TAG="comps"
13-
13+
export DATA_PATH=${model_cache}
1414
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
1515
source ${SCRIPT_DIR}/dataprep_utils.sh
1616

tests/dataprep/test_dataprep_neo4j_on_intel_hpu.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@ ip_address=$(hostname -I | awk '{print $1}')
1010
DATAPREP_PORT=11103
1111
LLM_ENDPOINT_PORT=10510
1212
export TAG="comps"
13+
export DATA_PATH=${model_cache}
1314

1415
function build_docker_images() {
1516
cd $WORKPATH

tests/dataprep/test_dataprep_qdrant.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@ ip_address=$(hostname -I | awk '{print $1}')
1010
DATAPREP_PORT="11107"
1111
TEI_EMBEDDER_PORT="10220"
1212
export TAG="comps"
13+
export DATA_PATH=${model_cache}
1314

1415
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
1516
source ${SCRIPT_DIR}/dataprep_utils.sh

tests/dataprep/test_dataprep_redis.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@ ip_address=$(hostname -I | awk '{print $1}')
1010
DATAPREP_PORT="11108"
1111
TEI_EMBEDDER_PORT="10221"
1212
export TAG="comps"
13+
export DATA_PATH=${model_cache}
1314

1415
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
1516
source ${SCRIPT_DIR}/dataprep_utils.sh

tests/dataprep/test_dataprep_redis_finance_on_intel_hpu.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ function start_vllm_service_70B() {
5353
echo "token is ${HF_TOKEN}"
5454
model="meta-llama/Llama-3.3-70B-Instruct"
5555
vllm_port=8086
56-
export HF_CACHE_DIR=/data2/hf_model
56+
export HF_CACHE_DIR=${model_cache:-./data}
5757
vllm_volume=$HF_CACHE_DIR
5858

5959
echo "start vllm gaudi service"

tests/dataprep/test_dataprep_redis_multimodal.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ caption_fn="${tmp_dir}/${image_name}.txt"
2323
pdf_name="nke-10k-2023"
2424
pdf_fn="${tmp_dir}/${pdf_name}.pdf"
2525
DATAPREP_PORT="11109"
26+
export DATA_PATH=${model_cache}
2627

2728
function build_docker_images() {
2829
cd $WORKPATH

0 commit comments

Comments
 (0)