TensorFlow .pb or ONNX Format

#38
by mariusmuntean - opened

How can I convert the .h5 file to .pb or .tflite? Alternatively .onnx would work too.
I'm a total beginner and googling around didn't yield anything that worked.

following

Hey @mariusmuntean and @0xrk , this model is already available in .h5!

See here

image.png

The recommended way to load this model in TensorFlow is with the following code:

>>> from transformers import TFAutoModelForSeq2SeqLM

>>> model = TFAutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn")
This comment has been hidden

Sign up or log in to comment