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