Skip to content

Commit 197678c

Browse files
fix CI test issue (#2310)
Signed-off-by: Yongbozzz <yongbo.zhu@intel.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent 760ff02 commit 197678c

File tree

5 files changed

+9
-2
lines changed

5 files changed

+9
-2
lines changed

EdgeCraftRAG/docker_compose/intel/gpu/arc/set_env.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,3 +34,4 @@ export MAX_NUM_BATCHED_TOKENS=${MAX_NUM_BATCHED_TOKENS}
3434
export MAX_MODEL_LEN=${MAX_MODEL_LEN}
3535
export LOAD_IN_LOW_BIT=${LOAD_IN_LOW_BIT}
3636
export CCL_DG2_USM=${CCL_DG2_USM}
37+
export ZE_AFFINITY_MASK=${ZE_AFFINITY_MASK}

EdgeCraftRAG/edgecraftrag/requirements.txt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,9 @@ pillow>=10.4.0
2222
py-cpuinfo>=9.0.0
2323
pymilvus==2.5.10
2424
python-docx==1.1.2
25+
torch==2.8.0+cpu
26+
torchvision==0.23.0+cpu
2527
unstructured
2628
unstructured[pdf]
2729
werkzeug==3.1.3
30+

EdgeCraftRAG/tests/test_compose_on_arc.sh

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,6 @@ UI_UPLOAD_PATH="$WORKPATH/tests"
2828

2929
HF_ENDPOINT=https://hf-mirror.com
3030

31-
3231
function build_docker_images() {
3332
opea_branch=${opea_branch:-"main"}
3433
cd $WORKPATH/docker_image_build

EdgeCraftRAG/tests/test_compose_vllm_on_arc.sh

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@ LLM_MODEL="Qwen/Qwen3-8B"
3838
LLM_MODEL_PATH="${MODEL_PATH}/${LLM_MODEL}"
3939
NGINX_CONFIG_PATH="$WORKPATH/nginx/nginx.conf"
4040
VLLM_IMAGE_TAG="0.8.3-b20"
41+
MAX_MODEL_LEN=8192
4142
DP_NUM=1
4243

4344
function build_docker_images() {
@@ -179,4 +180,5 @@ function main() {
179180

180181
}
181182

183+
182184
main

EdgeCraftRAG/tests/test_compose_vllm_on_arc_b60.sh

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,14 +25,15 @@ MODEL_PATH="${HOME}/models"
2525
# MODEL_PATH="$WORKPATH/models"
2626
DOC_PATH="$WORKPATH/tests"
2727
UI_UPLOAD_PATH="$WORKPATH/tests"
28-
28+
MAX_MODEL_LEN=8192
2929
HF_ENDPOINT=https://hf-mirror.com
3030
VLLM_SERVICE_PORT_B60=8086
3131
TP=1
3232
vLLM_ENDPOINT="http://${HOST_IP}:${VLLM_SERVICE_PORT_B60}"
3333
LLM_MODEL="Qwen/Qwen3-8B"
3434
VLLM_IMAGE_TAG="1.0"
3535
DP=1
36+
ZE_AFFINITY_MASK=1
3637

3738
function build_docker_images() {
3839
opea_branch=${opea_branch:-"main"}
@@ -170,4 +171,5 @@ function main() {
170171

171172
}
172173

174+
173175
main

0 commit comments

Comments
 (0)