From b46ae8bdcc1abfe04563cffc004a87d2884e111b Mon Sep 17 00:00:00 2001 From: yang yuhan Date: Wed, 24 Jul 2024 08:41:47 +0800 Subject: [PATCH] Fix a minor bug for chatqna in docker-compose (#442) (#443) Embedding and reranking services failed to run on GPU H100. Change the image tag and use CPU for these services. This PR will fix #442 Signed-off-by: PeterYang12 --- ChatQnA/docker/gpu/docker_compose.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ChatQnA/docker/gpu/docker_compose.yaml b/ChatQnA/docker/gpu/docker_compose.yaml index 15c2f3679..ee78c3346 100644 --- a/ChatQnA/docker/gpu/docker_compose.yaml +++ b/ChatQnA/docker/gpu/docker_compose.yaml @@ -27,7 +27,7 @@ services: REDIS_URL: ${REDIS_URL} INDEX_NAME: ${INDEX_NAME} tei-embedding-service: - image: ghcr.io/huggingface/text-embeddings-inference:1.2 + image: ghcr.io/huggingface/text-embeddings-inference:cpu-1.2 container_name: tei-embedding-server ports: - "8090:80" @@ -83,7 +83,7 @@ services: LANGCHAIN_PROJECT: "opea-retriever-service" restart: unless-stopped tei-reranking-service: - image: ghcr.io/huggingface/text-embeddings-inference:1.2 + image: ghcr.io/huggingface/text-embeddings-inference:cpu-1.2 container_name: tei-reranking-server ports: - "8808:80"