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

Optimisied base image

Browse Source
This commit is contained in:
Artsiom Siamashka
2025-11-18 15:40:26 +01:00
parent ef9dab7724
commit 41ec25afc1
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
MLBaseDockerfile
View File

@@ -1,14 +1,15 @@
FROM condaforge/miniforge3:25.9.1-0 FROM condaforge/miniforge3:25.9.1-0
WORKDIR /app WORKDIR /app
RUN mkdir /app/models RUN pip install huggingface_hub
# 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
RUN conda install -c conda-forge \ RUN conda install -c conda-forge \
numpy=1.26.4 \ numpy=1.26.4 \
pandas=2.3.3 \ pandas=2.3.3 \
sentence-transformers=3.0.1 \ sentence-transformers=3.0.1 \
keybert=0.9.0 \ keybert=0.9.0 \
faiss-gpu=1.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