optimum icon indicating copy to clipboard operation
optimum copied to clipboard

Published 20 hours ago • verified publisher icon huggingface

Add ONNX export for owlv2

Open xenova opened this issue 1 year ago • 4 comments

What does this PR do?

Fixes # (issue)

Before submitting

  • [ ] This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
  • [x] Did you make sure to update the documentation with your changes?
  • [x] Did you write any new necessary tests?

xenova avatar Feb 09 '24 14:02 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.

HuggingFaceDocBuilderDev avatar Feb 09 '24 14:02 HuggingFaceDocBuilderDev

Working model: https://huggingface.co/Xenova/owlv2-base-patch16-ensemble (+ tested with transformers.js)

cc @fxmarty

xenova avatar Feb 19 '24 12:02 xenova

@xenova Can you also rebase/merge main?

fxmarty avatar Feb 19 '24 13:02 fxmarty

Done! Turns out opset=14 isn't necessary 😇

xenova avatar Feb 19 '24 13:02 xenova

@xenova Could you either push an empty commit or merge main (there was an issue with pytest that is now pinned)?

Can you also confirm that the exported model behaves as expected?

fxmarty avatar Feb 26 '24 14:02 fxmarty

bump

fxmarty avatar Feb 29 '24 12:02 fxmarty

Done! (sorry for the delay; missed previous notification)

xenova avatar Feb 29 '24 14:02 xenova