Skip to content

Commit 781b533

Browse files
committed
fix tests
Signed-off-by: Artem Astafev <a.astafev@datamonsters.com>
1 parent 05f1eca commit 781b533

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

VisualQnA/docker_compose/amd/gpu/rocm/compose_vllm.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ services:
3030
security_opt:
3131
- seccomp:unconfined
3232
- apparmor=unconfined
33-
command: "--model ${VISUALQNA_LLM_MODEL_ID} --swap-space 16 --disable-log-requests --dtype float16 --tensor-parallel-size 1 --host 0.0.0.0 --port 8011 --num-scheduler-steps 1 --distributed-executor-backend \"mp\""
33+
command: "--model ${VISUALQNA_LVM_MODEL_ID} --swap-space 16 --disable-log-requests --dtype float16 --tensor-parallel-size 1 --host 0.0.0.0 --port 8011 --num-scheduler-steps 1 --distributed-executor-backend \"mp\""
3434
ipc: host
3535
lvm:
3636
image: ${REGISTRY:-opea}/lvm:${TAG:-latest}
@@ -46,7 +46,7 @@ services:
4646
https_proxy: ${https_proxy}
4747
LVM_COMPONENT_NAME: "OPEA_VLLM_LVM"
4848
LVM_ENDPOINT: ${LVM_ENDPOINT}
49-
LLM_MODEL_ID: ${VISUALQNA_LLM_MODEL_ID}
49+
LLM_MODEL_ID: ${VISUALQNA_LVM_MODEL_ID}
5050
HF_HUB_DISABLE_PROGRESS_BARS: 1
5151
HF_HUB_ENABLE_HF_TRANSFER: 0
5252
restart: unless-stopped

VisualQnA/docker_compose/amd/gpu/rocm/set_env_vllm.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ export VISUALQNA_VLLM_SERVICE_PORT="8081"
99
export VISUALQNA_HUGGINGFACEHUB_API_TOKEN=${Your_HUGGINGFACEHUB_API_TOKEN}
1010
export VISUALQNA_CARD_ID="card1"
1111
export VISUALQNA_RENDER_ID="renderD136"
12-
export VISUALQNA_LLM_MODEL_ID="Xkev/Llama-3.2V-11B-cot"
12+
export VISUALQNA_LVM_MODEL_ID="Xkev/Llama-3.2V-11B-cot"
1313
export LVM_ENDPOINT="http://${HOST_IP}:${VISUALQNA_VLLM_SERVICE_PORT}"
1414
export LVM_SERVICE_PORT=9399
1515
export MEGA_SERVICE_HOST_IP=${HOST_IP}

0 commit comments

Comments
 (0)