NhutP
/

ViWhisper-tiny

Automatic Speech Recognition

Model card Files Files and versions

NhutP commited on Apr 27, 2025

Commit

a3d2244

·

verified ·

1 Parent(s): f00a6d4

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-tiny")
 model = WhisperForConditionalGeneration.from_pretrained("NhutP/ViWhisper-tiny")
-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-tiny")
 model = WhisperForConditionalGeneration.from_pretrained("NhutP/ViWhisper-tiny")
+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