Added EXTRA_MODULES argument

This commit is contained in:
Jan-Timo Hesse
2024-08-05 09:42:16 +02:00
parent d3146d20ad
commit cec5fdd144
2 changed files with 7 additions and 1 deletions

View File

@@ -38,6 +38,7 @@ ARG USE_OLLAMA
ARG USE_CUDA_VER
ARG USE_EMBEDDING_MODEL
ARG USE_RERANKING_MODEL
ARG EXTRA_MODULES
ARG UID
ARG GID
@@ -49,7 +50,8 @@ ENV ENV=prod \
USE_CUDA_DOCKER=${USE_CUDA} \
USE_CUDA_DOCKER_VER=${USE_CUDA_VER} \
USE_EMBEDDING_MODEL_DOCKER=${USE_EMBEDDING_MODEL} \
USE_RERANKING_MODEL_DOCKER=${USE_RERANKING_MODEL}
USE_RERANKING_MODEL_DOCKER=${USE_RERANKING_MODEL} \
EXTRA_MODULES_DOCKER=${EXTRA_MODULES}
## Basis URL Config ##
ENV OLLAMA_BASE_URL="/ollama" \