Skip to content

Commit 5da8fa2

Browse files
committed
Bake BGE embedding model into API image
1 parent fe321f3 commit 5da8fa2

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

api/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends gcc libpq-dev &
77
COPY requirements.txt .
88
RUN pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cpu torch==2.3.1
99
RUN pip install --no-cache-dir -r requirements.txt
10-
RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('all-MiniLM-L6-v2')"
10+
RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('BAAI/bge-small-en-v1.5')"
1111

1212
COPY . /app/api
1313

0 commit comments

Comments
 (0)