For example, for DistilBERT we have: | |
thon | |
from transformers.models.distilbert import DistilBertConfig, DistilBertOnnxConfig | |
config = DistilBertConfig() | |
onnx_config = DistilBertOnnxConfig(config) | |
print(list(onnx_config.outputs.keys())) | |
["last_hidden_state"] | |
The process is identical for TensorFlow checkpoints on the Hub. |