scream_medium_beta / convert_to_pytorch.py
pere's picture
PyTorch models converted
7307350
raw
history blame contribute delete
272 Bytes
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.")