lydgs commited on
Commit
bb89db8
·
verified ·
1 Parent(s): ab0708a

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +11 -7
Dockerfile CHANGED
@@ -1,13 +1,17 @@
 
1
  FROM michaelf34/infinity:latest
2
 
3
- # Hugging Face 缓存目录设到可写空间,避免模型被重复下载
 
 
 
4
  ENV HF_HOME=/data/huggingface
5
 
6
- # 通过环境变量指定要加载的模型 ID
 
7
  ENV INFINITY_MODEL_ID="sentence-transformers/all-MiniLM-L6-v2"
8
 
9
- # 明确指定使用 CPU 引擎,适配 HF 的免费环境
10
- ENV INFINITY_ENGINE="torch"
11
-
12
- # 使用 Infinity 自带的 CLI 启动,并指定端口为 7860
13
- CMD ["infinity_emb", "v2", "--port", "7860"]
 
1
+ # 使用 Infinity 官方最新 CPU 镜像
2
  FROM michaelf34/infinity:latest
3
 
4
+ # --- 国内网络环境加速配 ---
5
+ # 设置 Hugging Face 国内镜像,解决模型下载难题
6
+ ENV HF_ENDPOINT=https://hf-mirror.com
7
+ # 指定模型缓存目录到可写空间,避免重复下载
8
  ENV HF_HOME=/data/huggingface
9
 
10
+ # --- 模型选择 ---
11
+ # 通过环境变量指定需要加载的模型,推荐使用轻量模型以匹配免费 CPU 资源
12
  ENV INFINITY_MODEL_ID="sentence-transformers/all-MiniLM-L6-v2"
13
 
14
+ # --- 启动命令修正 ---
15
+ # 关键:最新版镜像正确启动方式是用 `infinity_emb v2` 命令并配合 `--port 7860` 参数
16
+ # 这样会监听 Hugging Face 要求的 7860 端口
17
+ CMD ["infinity_emb", "v2", "--model-id", "sentence-transformers/all-MiniLM-L6-v2", "--port", "7860"]