Fix a minor bug for chatqna in docker-compose (opea-project#442) (ope…

…a-project#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 opea-project#442 Signed-off-by: PeterYang12 <[email protected]>
lvliang-intel · Jul 24, 2024 · b46ae8b · b46ae8b
1 parent f45f508
commit b46ae8b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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"