Unverified Commit 5fa0a1b2 authored by Sadra Barikbin's avatar Sadra Barikbin Committed by GitHub
Browse files

Fix a tiny typo in `WhisperForConditionalGeneration::generate` docstring (#24045)

parent ba695c1e
...@@ -1519,10 +1519,10 @@ class WhisperForConditionalGeneration(WhisperPreTrainedModel): ...@@ -1519,10 +1519,10 @@ class WhisperForConditionalGeneration(WhisperPreTrainedModel):
Whether to continue running the while loop until max_length (needed for ZeRO stage 3) Whether to continue running the while loop until max_length (needed for ZeRO stage 3)
return_timestamps (`bool`, *optional*): return_timestamps (`bool`, *optional*):
Whether to return the timestamps with the text. This enables the `WhisperTimestampsLogitsProcessor`. Whether to return the timestamps with the text. This enables the `WhisperTimestampsLogitsProcessor`.
task (`bool`, *optional*): task (`str`, *optional*):
Task to use for generation, either "translate" or "transcribe". The `model.config.forced_decoder_ids` Task to use for generation, either "translate" or "transcribe". The `model.config.forced_decoder_ids`
will be updated accordingly. will be updated accordingly.
language (`bool`, *optional*): language (`str`, *optional*):
Language token to use for generation, can be either in the form of `<|en|>`, `en` or `english`. You can Language token to use for generation, can be either in the form of `<|en|>`, `en` or `english`. You can
find all the possible language tokens in the `model.generation_config.lang_to_id` dictionary. find all the possible language tokens in the `model.generation_config.lang_to_id` dictionary.
is_multilingual (`bool`, *optional*): is_multilingual (`bool`, *optional*):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment