jandres/GenAIExamples
1
0
Fork 0
Code Issues Pull Requests Actions 5 Packages Projects Releases Wiki Activity

Enable GraphRAG and ProductivitySuite model cache for docker compose test. (#1608)

Browse Source 
Signed-off-by: ZePan110 <ze.pan@intel.com>
This commit is contained in:
ZePan110
2025-03-13 11:23:03 +08:00
committed by GitHub
parent 0701b8cfff
commit d670dbf0aa
4 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
GraphRAG/docker_compose/intel/hpu/gaudi/compose.yaml
View File

@@ -39,7 +39,7 @@ services:
ports: ports:
- "${TEI_EMBEDDER_PORT:-12000}:80" - "${TEI_EMBEDDER_PORT:-12000}:80"
volumes: volumes:
- "./data:/data" - "${MODEL_CACHE:-./data}:/data"
shm_size: 1g shm_size: 1g
environment: environment:
no_proxy: ${no_proxy} no_proxy: ${no_proxy}
@@ -58,7 +58,7 @@ services:
ports: ports:
- ${LLM_ENDPOINT_PORT:-8008}:80 - ${LLM_ENDPOINT_PORT:-8008}:80
volumes: volumes:
- "${DATA_PATH:-./data}:/data" - "${MODEL_CACHE:-./data}:/data"
environment: environment:
no_proxy: ${no_proxy} no_proxy: ${no_proxy}
http_proxy: ${http_proxy} http_proxy: ${http_proxy}

1
GraphRAG/tests/test_compose_on_gaudi.sh
View File

@@ -9,6 +9,7 @@ echo "REGISTRY=IMAGE_REPO=${IMAGE_REPO}"
echo "TAG=IMAGE_TAG=${IMAGE_TAG}" echo "TAG=IMAGE_TAG=${IMAGE_TAG}"
export REGISTRY=${IMAGE_REPO} export REGISTRY=${IMAGE_REPO}
export TAG=${IMAGE_TAG} export TAG=${IMAGE_TAG}
export MODEL_CACHE=${model_cache:-"./data"}
WORKPATH=$(dirname "$PWD") WORKPATH=$(dirname "$PWD")
LOG_PATH="$WORKPATH/tests" LOG_PATH="$WORKPATH/tests"

8
ProductivitySuite/docker_compose/intel/cpu/xeon/compose.yaml
View File

@@ -39,7 +39,7 @@ services:
ports: ports:
- "6006:80" - "6006:80"
volumes: volumes:
- "./data_embedding:/data" - "${MODEL_CACHE:-./data_embedding}:/data"
shm_size: 1g shm_size: 1g
environment: environment:
no_proxy: ${no_proxy} no_proxy: ${no_proxy}
@@ -95,7 +95,7 @@ services:
ports: ports:
- "8808:80" - "8808:80"
volumes: volumes:
- "./data_tei:/data" - "${MODEL_CACHE:-./data_tei}:/data"
shm_size: 1g shm_size: 1g
environment: environment:
no_proxy: ${no_proxy} no_proxy: ${no_proxy}
@@ -136,7 +136,7 @@ services:
ports: ports:
- "9009:80" - "9009:80"
volumes: volumes:
- "./data:/data" - "${MODEL_CACHE:-./data}:/data"
shm_size: 1g shm_size: 1g
environment: environment:
no_proxy: ${no_proxy} no_proxy: ${no_proxy}
@@ -209,7 +209,7 @@ services:
ports: ports:
- "8028:80" - "8028:80"
volumes: volumes:
- "./data_codegen:/data" - "${MODEL_CACHE:-./data_codegen}:/data"
shm_size: 1g shm_size: 1g
environment: environment:
no_proxy: ${no_proxy} no_proxy: ${no_proxy}

1
ProductivitySuite/tests/test_compose_on_xeon.sh
View File

@@ -9,6 +9,7 @@ echo "REGISTRY=IMAGE_REPO=${IMAGE_REPO}"
echo "TAG=IMAGE_TAG=${IMAGE_TAG}" echo "TAG=IMAGE_TAG=${IMAGE_TAG}"
export REGISTRY=${IMAGE_REPO} export REGISTRY=${IMAGE_REPO}
export TAG=${IMAGE_TAG} export TAG=${IMAGE_TAG}
export MODEL_CACHE=${model_cache:-"./data"}
WORKPATH=$(dirname "$PWD") WORKPATH=$(dirname "$PWD")
LOG_PATH="$WORKPATH/tests" LOG_PATH="$WORKPATH/tests"