Update Dockerfile
Browse files- Dockerfile +3 -1
Dockerfile
CHANGED
|
@@ -5,7 +5,9 @@ ENV HF_ENDPOINT=https://hf-mirror.com
|
|
| 5 |
ENV HF_HOME=/data/huggingface
|
| 6 |
|
| 7 |
# 模型选择（轻量模型，适合免费 CPU）
|
| 8 |
-
ENV INFINITY_MODEL_ID="microsoft/harrier-oss-v1-270m"
|
|
|
|
|
|
|
| 9 |
|
| 10 |
# 注意：CMD 中不要重复写 infinity_emb，直接写子命令和参数
|
| 11 |
CMD ["v2", "--model-id", "sentence-transformers/all-MiniLM-L6-v2", "--port", "7860"]
|
|
|
|
| 5 |
ENV HF_HOME=/data/huggingface
|
| 6 |
|
| 7 |
# 模型选择（轻量模型，适合免费 CPU）
|
| 8 |
+
# ENV INFINITY_MODEL_ID="microsoft/harrier-oss-v1-270m"
|
| 9 |
+
|
| 10 |
+
ENV INFINITY_MODEL_ID="microsoft/harrier-oss-v1-0.6b"
|
| 11 |
|
| 12 |
# 注意：CMD 中不要重复写 infinity_emb，直接写子命令和参数
|
| 13 |
CMD ["v2", "--model-id", "sentence-transformers/all-MiniLM-L6-v2", "--port", "7860"]
|