This commit is contained in:
Timothy J. Baek
2024-09-19 21:51:45 +02:00
parent 2dad9b9432
commit cd117f5b67
2 changed files with 4 additions and 12 deletions

View File

@@ -74,6 +74,10 @@ ENV RAG_EMBEDDING_MODEL="$USE_EMBEDDING_MODEL_DOCKER" \
## Hugging Face download cache ##
ENV HF_HOME="/app/backend/data/cache/embedding/models"
## Torch Extensions ##
ENV TORCH_EXTENSIONS_DIR="/app/backend/data/cache/torch_extensions"
#### Other models ##########################################################
WORKDIR /app/backend