optimum icon indicating copy to clipboard operation
optimum copied to clipboard

[WIP] Add onnx export support for CLIPSeg models

Open xenova opened this issue 1 year ago • 1 comments

What does this PR do?

Two things need to be updated in transformers:

  1. Fix casting here, here, and here
  2. 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?

xenova avatar Dec 25 '23 22:12 xenova

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.