kamp0010 commited on
Commit
b6ccdfc
·
verified ·
1 Parent(s): be98905

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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/model_fp16.onnx', 'provider': 'CPUExecutionProvider'}, \
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 \