Skip to content

Add MobileBert to NormalizedConfigManager - #2490

Open
yomidotk wants to merge 1 commit into
huggingface:mainfrom
yomidotk:add-mobilebert-config
Open

yomidotk wants to merge 1 commit into
huggingface:mainfrom
yomidotk:add-mobilebert-config

Conversation

@yomidotk

@yomidotk yomidotk commented Sep 7, 2026

Copy link
Copy Markdown

What does this PR do?

This PR adds MobileBert to the NormalizedConfigManager to help extend ONNX optimization support.

Changes:

  • Mapped mobilebert to NormalizedTextConfig in optimum/utils/normalized_config.py.
  • Removed mobilebert from the missing configs TODO list.
  • Added a test case for mobilebert in TEXT_ENCODER_MODELS within tests/utils/test_dummpy_input_generators.py.

All local tests passed successfully.

Fixes #1272

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?

Who can review?

@echarlaix @JingyaHuang @michaelbenayoun @IlyasMoutawwakil

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.

1 participant