Skip to content

Commit 7c08862

Browse files
committed
fix multimodal retriever issues
Signed-off-by: letonghan <letong.han@intel.com>
1 parent 9a61e5a commit 7c08862

File tree

5 files changed

+5
-1
lines changed

5 files changed

+5
-1
lines changed

MultimodalQnA/docker_compose/intel/cpu/xeon/compose.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,7 @@ services:
8888
https_proxy: ${https_proxy}
8989
REDIS_URL: ${REDIS_URL}
9090
INDEX_NAME: ${INDEX_NAME}
91+
BRIDGE_TOWER_EMBEDDING: ${BRIDGE_TOWER_EMBEDDING}
9192
restart: unless-stopped
9293
lvm-llava:
9394
image: ${REGISTRY:-opea}/lvm-llava:${TAG:-latest}

MultimodalQnA/docker_compose/intel/hpu/gaudi/compose.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,7 @@ services:
8888
https_proxy: ${https_proxy}
8989
REDIS_URL: ${REDIS_URL}
9090
INDEX_NAME: ${INDEX_NAME}
91+
BRIDGE_TOWER_EMBEDDING: ${BRIDGE_TOWER_EMBEDDING}
9192
restart: unless-stopped
9293
tgi-gaudi:
9394
image: ghcr.io/huggingface/tgi-gaudi:2.0.6

MultimodalQnA/multimodalqna.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ def add_remote_service(self):
5454
name="retriever",
5555
host=MM_RETRIEVER_SERVICE_HOST_IP,
5656
port=MM_RETRIEVER_SERVICE_PORT,
57-
endpoint="/v1/multimodal_retrieval",
57+
endpoint="/v1/retrieval",
5858
use_remote_service=True,
5959
service_type=ServiceType.RETRIEVER,
6060
)

MultimodalQnA/tests/test_compose_on_gaudi.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ function setup_env() {
4141
export REDIS_URL="redis://${host_ip}:6379"
4242
export REDIS_HOST=${host_ip}
4343
export INDEX_NAME="mm-rag-redis"
44+
export BRIDGE_TOWER_EMBEDDING=true
4445
export LLAVA_SERVER_PORT=8399
4546
export LVM_ENDPOINT="http://${host_ip}:8399"
4647
export EMBEDDING_MODEL_ID="BridgeTower/bridgetower-large-itm-mlm-itc"

MultimodalQnA/tests/test_compose_on_xeon.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@ function setup_env() {
3939
export REDIS_URL="redis://${host_ip}:6379"
4040
export REDIS_HOST=${host_ip}
4141
export INDEX_NAME="mm-rag-redis"
42+
export BRIDGE_TOWER_EMBEDDING=true
4243
export LLAVA_SERVER_PORT=8399
4344
export LVM_ENDPOINT="http://${host_ip}:8399"
4445
export LVM_MODEL_ID="llava-hf/llava-1.5-7b-hf"

0 commit comments

Comments
 (0)