Skip to content

Commit 1b30783

Browse files
authored
Add key TEI and TGI parameters for handling long retrievals (#340)
1 parent 2e62ecc commit 1b30783

File tree

4 files changed

+6
-6
lines changed

4 files changed

+6
-6
lines changed

ChatQnA/docker/gaudi/docker_compose.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ services:
9292
HUGGINGFACEHUB_API_TOKEN: ${HUGGINGFACEHUB_API_TOKEN}
9393
HF_HUB_DISABLE_PROGRESS_BARS: 1
9494
HF_HUB_ENABLE_HF_TRANSFER: 0
95-
command: --model-id ${RERANK_MODEL_ID}
95+
command: --model-id ${RERANK_MODEL_ID} --auto-truncate
9696
reranking:
9797
image: opea/reranking-tei:latest
9898
container_name: reranking-tei-gaudi-server

ChatQnA/docker/xeon/docker_compose.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ services:
3636
no_proxy: ${no_proxy}
3737
http_proxy: ${http_proxy}
3838
https_proxy: ${https_proxy}
39-
command: --model-id ${EMBEDDING_MODEL_ID}
39+
command: --model-id ${EMBEDDING_MODEL_ID} --auto-truncate
4040
embedding:
4141
image: opea/embedding-tei:latest
4242
container_name: embedding-tei-server
@@ -88,7 +88,7 @@ services:
8888
HUGGINGFACEHUB_API_TOKEN: ${HUGGINGFACEHUB_API_TOKEN}
8989
HF_HUB_DISABLE_PROGRESS_BARS: 1
9090
HF_HUB_ENABLE_HF_TRANSFER: 0
91-
command: --model-id ${RERANK_MODEL_ID}
91+
command: --model-id ${RERANK_MODEL_ID} --auto-truncate
9292
reranking:
9393
image: opea/reranking-tei:latest
9494
container_name: reranking-tei-xeon-server

SearchQnA/docker/gaudi/docker_compose.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ services:
6666
no_proxy: ${no_proxy}
6767
http_proxy: ${http_proxy}
6868
https_proxy: ${https_proxy}
69-
command: --model-id ${RERANK_MODEL_ID}
69+
command: --model-id ${RERANK_MODEL_ID} --auto-truncate
7070
reranking:
7171
image: opea/reranking-tei:latest
7272
container_name: reranking-tei-xeon-server

SearchQnA/docker/xeon/docker_compose.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ services:
1717
no_proxy: ${no_proxy}
1818
http_proxy: ${http_proxy}
1919
https_proxy: ${https_proxy}
20-
command: --model-id ${EMBEDDING_MODEL_ID}
20+
command: --model-id ${EMBEDDING_MODEL_ID} --auto-truncate
2121
embedding:
2222
image: opea/embedding-tei:latest
2323
container_name: embedding-tei-server
@@ -62,7 +62,7 @@ services:
6262
no_proxy: ${no_proxy}
6363
http_proxy: ${http_proxy}
6464
https_proxy: ${https_proxy}
65-
command: --model-id ${RERANK_MODEL_ID}
65+
command: --model-id ${RERANK_MODEL_ID} --auto-truncate
6666
reranking:
6767
image: opea/reranking-tei:latest
6868
container_name: reranking-tei-xeon-server

0 commit comments

Comments
 (0)