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
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")