Update Dockerfile
Browse files- Dockerfile +11 -7
Dockerfile
CHANGED
|
@@ -1,13 +1,17 @@
|
|
|
|
|
| 1 |
FROM michaelf34/infinity:latest
|
| 2 |
|
| 3 |
-
#
|
|
|
|
|
|
|
|
|
|
| 4 |
ENV HF_HOME=/data/huggingface
|
| 5 |
|
| 6 |
-
#
|
|
|
|
| 7 |
ENV INFINITY_MODEL_ID="sentence-transformers/all-MiniLM-L6-v2"
|
| 8 |
|
| 9 |
-
#
|
| 10 |
-
|
| 11 |
-
|
| 12 |
-
|
| 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"]
|
|
|