optimum
optimum copied to clipboard
[WIP] Add onnx export support for CLIPSeg models
What does this PR do?
Two things need to be updated in transformers:
- Fix casting here, here, and here
- Add the following line here:
("clipseg", "CLIPSegForImageSegmentation"),
Fixes # (issue)
Before submitting
- [ ] This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
- [ ] Did you make sure to update the documentation with your changes?
- [ ] Did you write any new necessary tests?
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.