Optimum documentation

Adding support for an unsupported architecture

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v1.23.3).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

Adding support for an unsupported architecture

If you wish to export a model whose architecture is not already supported by the library, the PR #813 Adds support for ResNet can be used as a reference.

You can make sure tests pass for the new my_new_modeltype model type by running:

pytest tests/exporters/tflite/test_*.py -k "my_new_modeltype" -s --exitfirst
< > Update on GitHub