Skip to content

Rembert onnx support#2108

Merged
echarlaix merged 8 commits into
huggingface:mainfrom
mlynatom:rembert-onnx-support
Dec 2, 2024
Merged

Rembert onnx support#2108
echarlaix merged 8 commits into
huggingface:mainfrom
mlynatom:rembert-onnx-support

Conversation

@mlynatom

@mlynatom mlynatom commented Nov 30, 2024

Copy link
Copy Markdown
Contributor

What does this PR do?

Fixes #2092

Hi, this pull request adds support for the RemBERT model based on the example mentioned in issue #2092.

Before submitting

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

Who can review?

@IlyasMoutawwakil

@mlynatom mlynatom marked this pull request as ready for review November 30, 2024 14:46
@HuggingFaceDocBuilderDev

Copy link
Copy Markdown

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.

Comment thread optimum/exporters/tasks.py Outdated
Comment thread optimum/exporters/tasks.py
Comment thread tests/onnxruntime/test_modeling.py Outdated

@echarlaix echarlaix left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks for your contribution @mlynatom !

@echarlaix echarlaix merged commit f22655c into huggingface:main Dec 2, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Add support for RemBERT in the ONNX export

4 participants