BG5 commited on
Commit
c73c40b
·
verified ·
1 Parent(s): 291c23f

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -5,7 +5,7 @@ USER user
5
 
6
  # 环境变量设置
7
  ENV RAG_EMBEDDING_MODEL="sentence-transformers/all-MiniLM-L6-v2" \
8
- SENTENCE_TRANSFORMERS_HOME="/app/models" \
9
  RAG_EMBEDDING_MODEL_AUTO_UPDATE="true" \
10
  PATH="/home/user/.local/bin:$PATH" \
11
  DATA_DIR="/app/data" \
@@ -15,6 +15,7 @@ WORKDIR /app
15
  RUN mkdir -p $SENTENCE_TRANSFORMERS_HOME
16
  # 安装应用和依赖
17
  RUN pip install --no-cache-dir open-webui
 
18
  RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('$RAG_EMBEDDING_MODEL')"
19
  # 添加数据同步脚本
20
  COPY --chown=user . /app
 
5
 
6
  # 环境变量设置
7
  ENV RAG_EMBEDDING_MODEL="sentence-transformers/all-MiniLM-L6-v2" \
8
+ SENTENCE_TRANSFORMERS_HOME="/app/data/models" \
9
  RAG_EMBEDDING_MODEL_AUTO_UPDATE="true" \
10
  PATH="/home/user/.local/bin:$PATH" \
11
  DATA_DIR="/app/data" \
 
15
  RUN mkdir -p $SENTENCE_TRANSFORMERS_HOME
16
  # 安装应用和依赖
17
  RUN pip install --no-cache-dir open-webui
18
+ RUN pip install "huggingface_hub[hf_xet]"
19
  RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('$RAG_EMBEDDING_MODEL')"
20
  # 添加数据同步脚本
21
  COPY --chown=user . /app