Optimum documentation

Adding support for an unsupported architecture

You are viewing v1.14.0 version. A newer version v1.17.1 is available.
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