NhutP
/

ViWhisper-base

Automatic Speech Recognition

Model card Files Files and versions

NhutP commited on Apr 27

Commit

c924c99

·

verified ·

1 Parent(s): 5d7e4d7

Update README.md

Files changed (1) hide show

README.md +4 -1

README.md CHANGED Viewed

@@ -38,7 +38,10 @@ import librosa
 # load model and processor
 processor = WhisperProcessor.from_pretrained("NhutP/ViWhisper-base")
 model = WhisperForConditionalGeneration.from_pretrained("NhutP/ViWhisper-base")
-model.config.forced_decoder_ids = None
 # load a sample
 array, sampling_rate = librosa.load('path_to_audio', sr = 16000) # Load some audio sample

 # load model and processor
 processor = WhisperProcessor.from_pretrained("NhutP/ViWhisper-base")
 model = WhisperForConditionalGeneration.from_pretrained("NhutP/ViWhisper-base")
+prefix_ids = model.generation_config.forced_decoder_ids
+model.generation_config.input_ids = prefix_ids
+model.generation_config.forced_decoder_ids = None
 # load a sample
 array, sampling_rate = librosa.load('path_to_audio', sr = 16000) # Load some audio sample