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

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -10
Dockerfile CHANGED
@@ -1,17 +1,11 @@
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"]
 
 
1
  FROM michaelf34/infinity:latest
2
 
3
+ # 国内加速(可选)
 
4
  ENV HF_ENDPOINT=https://hf-mirror.com
 
5
  ENV HF_HOME=/data/huggingface
6
 
7
+ # 模型选择(轻量模型,适合免费 CPU)
 
8
  ENV INFINITY_MODEL_ID="sentence-transformers/all-MiniLM-L6-v2"
9
 
10
+ # 注意:CMD 中不要重复写 infinity_emb,直接写子命令和参数
11
+ CMD ["v2", "--model-id", "sentence-transformers/all-MiniLM-L6-v2", "--port", "7860"]