Update Dockerfile
Browse files- 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 |
-
|
| 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"]
|
|
|
|
|
|