optimum
0f5acebb - Fix SAM ONNX export requirements with transformers 4.32, export vision encoder separately (#1301)

Commit
2 years ago
Fix SAM ONNX export requirements with transformers 4.32, export vision encoder separately (#1301) * support separate sam vision encoder * style * add test * Update optimum/exporters/onnx/__main__.py Co-authored-by: Michael Benayoun <mickbenayoun@gmail.com> * update on review * Update optimum/exporters/onnx/base.py Co-authored-by: Michael Benayoun <mickbenayoun@gmail.com> * update docstring * nit * fix * fix test * fix merge mess * fix * rename prompt encoder - mask decoder * unwanted change --------- Co-authored-by: Michael Benayoun <mickbenayoun@gmail.com>
Author
Parents
Loading