Skip to content

Benchmark: Add LayoutLMv2 to ONNX export benchmark#44709

Open
champ24-36 wants to merge 1 commit intohuggingface:mainfrom
champ24-36:bot-fix-1773512640247
Open

Benchmark: Add LayoutLMv2 to ONNX export benchmark#44709
champ24-36 wants to merge 1 commit intohuggingface:mainfrom
champ24-36:bot-fix-1773512640247

Conversation

@champ24-36
Copy link

This PR adds LayoutLMv2 to the existing ONNX export benchmark suite. This addresses issue #14368 and ensures that future changes don't inadvertently break ONNX export functionality for this model.

Specifically, it adds a new job to the benchmark workflow that:

  • Installs necessary dependencies for LayoutLMv2.
  • Downloads a pre-trained LayoutLMv2 model.
  • Exports the model to ONNX format.
  • Verifies the successful export (basic check).

This provides a baseline for performance and correctness of ONNX export for LayoutLMv2.


Contributed via automated PR bot

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