NbAiLabBeta
/

nb-whisper-small

Automatic Speech Recognition

hf-asr-leaderboard

Model card Files Files and versions

Metrics Training metrics Community

pere commited on Jan 11, 2024

Commit

26addf8

·

1 Parent(s): 966f7fa

Update export_models.sh

Files changed (1) hide show

export_models.sh +1 -1

export_models.sh CHANGED Viewed

@@ -12,7 +12,7 @@ shutil.copyfile('./generation_config.json', './generation_config_backup.json')
 print("Saving model to PyTorch...", end=" ")
 model = WhisperForConditionalGeneration.from_pretrained("./", from_flax=True)
 model.save_pretrained("./", safe_serialization=True)
-model.save_pretrained("./")
 print("Done.")
 print("Saving model to TensorFlow...", end=" ")

 print("Saving model to PyTorch...", end=" ")
 model = WhisperForConditionalGeneration.from_pretrained("./", from_flax=True)
 model.save_pretrained("./", safe_serialization=True)
+model.save_pretrained("./", safe_serialization=False)
 print("Done.")
 print("Saving model to TensorFlow...", end=" ")