pere's picture
converted to pytorch
33aed67
from transformers import WhisperForConditionalGeneration
model = WhisperForConditionalGeneration.from_pretrained(".", from_flax=True)
model.save_pretrained(".")
print("\nModel is now converted to PyTorch. There should be a new pytorch_model.bin-file in this directory.")