Update Dockerfile
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
|
@@ -26,10 +26,10 @@ from sentence_transformers import SentenceTransformer; \
|
|
| 26 |
SentenceTransformer( \
|
| 27 |
'jinaai/jina-embeddings-v2-base-code', \
|
| 28 |
backend='onnx', \
|
| 29 |
-
model_kwargs={'file_name': 'onnx/
|
| 30 |
trust_remote_code=True, \
|
| 31 |
); \
|
| 32 |
-
print('ONNX model cached.')" || echo "Pre-cache skipped — will download at runtime."
|
| 33 |
|
| 34 |
USER user
|
| 35 |
ENV HOME=/home/user \
|
|
|
|
| 26 |
SentenceTransformer( \
|
| 27 |
'jinaai/jina-embeddings-v2-base-code', \
|
| 28 |
backend='onnx', \
|
| 29 |
+
model_kwargs={'file_name': 'onnx/model.onnx', 'provider': 'CPUExecutionProvider'}, \
|
| 30 |
trust_remote_code=True, \
|
| 31 |
); \
|
| 32 |
+
print('ONNX fp32 model cached.')" || echo "Pre-cache skipped — will download at runtime."
|
| 33 |
|
| 34 |
USER user
|
| 35 |
ENV HOME=/home/user \
|