NbAiLabArchive
/

scream_small_beta

Automatic Speech Recognition

hf-asr-leaderboard

Model card Files Files and versions

Metrics Training metrics Community

scream_small_beta / convert_to_pytorch.py

pere's picture

PyTorch model converted

ae05e12 about 3 years ago

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