lxbfYeaa/AIServicesBaseImages
Archived
Template
1
0
Fork 0
forked from LiveCarta/AIServicesBaseImages
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added required models

Browse Source
This commit is contained in:
Artsiom Siamashka
2025-11-18 15:13:42 +01:00
parent 4d785fb805
commit ef9dab7724
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
MLBaseDockerfile
View File

@@ -1,8 +1,14 @@
FROM condaforge/miniforge3:25.9.1-0
WORKDIR /app
RUN mkdir /app/models
RUN conda install -c conda-forge \
numpy=1.26.4 \
pandas=2.3.3 \
sentence-transformers=3.0.1 \
keybert=0.9.0 \
faiss-gpu=1.9.0
# download cross-encoder and sentence-transformer models to cache
RUN hf download sentence-transformers/all-distilroberta-v1 --local-dir /assets/models/all-distilroberta-v1
RUN hf download cross-encoder/qnli-distilroberta-base --local-dir /assets/models/qnli-distilroberta-base