ssonpull519 commited on
Commit
85a2e68
·
verified ·
1 Parent(s): 55c27f2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -1
README.md CHANGED
@@ -60,7 +60,13 @@ print(scores)
60
 
61
  ## ONNX Conversion Script
62
 
63
- To use `optimum-cli`, `ColPaliForRetrieval` in transformers should be mapped in `AutoModel`, but currently is not. So I just sticked to this script, which is based on the official [guide](https://huggingface.co/docs/optimum/en/exporters/onnx/usage_guides/contribute#exporting-the-model) from huggingface.
 
 
 
 
 
 
64
 
65
  ```python
66
  from pathlib import Path
 
60
 
61
  ## ONNX Conversion Script
62
 
63
+ Currently in [pull request](https://github.com/huggingface/optimum/pull/2251) (not merged yet).
64
+
65
+ ```bash
66
+ optimum-cli export onnx --model vidore/colpali-v1.3-hf ./onnx_output --task feature-extraction --variant text --dtype fp16
67
+ ```
68
+
69
+ For fp16, there's an issue with transformers that is not fixed for now, so please use script below.
70
 
71
  ```python
72
  from pathlib import Path