File size: 272 Bytes
ae05e12 |
1 2 3 4 5 6 |
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.")
|