neurlang commited on
Commit
949f7c4
·
verified ·
1 Parent(s): 9f389d6

Update README.md

Browse files

Fixed first code snippet

Files changed (1) hide show
  1. README.md +3 -0
README.md CHANGED
@@ -251,6 +251,9 @@ In this example, the context tokens are 'unforced', meaning the model automatica
251
  >>> processor = WhisperProcessor.from_pretrained("neurlang/ipa-whisper-base")
252
  >>> model = WhisperForConditionalGeneration.from_pretrained("neurlang/ipa-whisper-base")
253
  >>> model.config.forced_decoder_ids = None
 
 
 
254
 
255
  >>> # load dummy dataset and read audio files
256
  >>> ds = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")
 
251
  >>> processor = WhisperProcessor.from_pretrained("neurlang/ipa-whisper-base")
252
  >>> model = WhisperForConditionalGeneration.from_pretrained("neurlang/ipa-whisper-base")
253
  >>> model.config.forced_decoder_ids = None
254
+ >>> model.config.suppress_tokens = []
255
+ >>> model.generation_config.forced_decoder_ids = None
256
+ >>> model.generation_config._from_model_config = True
257
 
258
  >>> # load dummy dataset and read audio files
259
  >>> ds = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")