Download new model for the emebdding (#24)

Signed-off-by: Alex Co <alex.tuan@mindvalley.com>
mindvalley · Apr 26, 2024 · 0eda4df · 0eda4df
1 parent f5dce11
commit 0eda4df
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/backend/Dockerfile.model_server b/backend/Dockerfile.model_server
@@ -23,8 +23,10 @@ from huggingface_hub import snapshot_download; \
 AutoTokenizer.from_pretrained('danswer/intent-model'); \
 AutoTokenizer.from_pretrained('intfloat/e5-base-v2'); \
 AutoTokenizer.from_pretrained('mixedbread-ai/mxbai-rerank-xsmall-v1'); \
+AutoTokenizer.from_pretrained('intfloat/e5-mistral-7b-instruct'); \
 snapshot_download('danswer/intent-model'); \
 snapshot_download('intfloat/e5-base-v2'); \
+snapshot_download('intfloat/e5-mistral-7b-instruct'); \
 snapshot_download('mixedbread-ai/mxbai-rerank-xsmall-v1')"
 
 WORKDIR /app