forked from LiveCarta/AIServicesBaseImages
17 lines
682 B
Plaintext
17 lines
682 B
Plaintext
FROM condaforge/miniforge3:25.9.1-0
|
|
|
|
WORKDIR /app
|
|
RUN pip install --root-user-action ignore huggingface_hub
|
|
|
|
# download cross-encoder and sentence-transformer models to cache
|
|
RUN hf download sentence-transformers/all-distilroberta-v1 --local-dir /assets/models/sentence-transformers/all-distilroberta-v1
|
|
RUN hf download cross-encoder/qnli-distilroberta-base --local-dir /assets/models/cross-encoder/qnli-distilroberta-base
|
|
|
|
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
|
|
RUN pip install --root-user-action ignore torch torchvision --index-url https://download.pytorch.org/whl/cu130
|