scream_small_beta / convert_to_pytorch.py
pere's picture
PyTorch model converted
ae05e12
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.")