mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-03 12:50:06 +06:00
Proper formatting
This commit is contained in:
parent
09bbaf93e7
commit
ccbae156f2
@ -699,7 +699,7 @@ class GenerationMixin:
|
||||
elif self.config.model_type in ["whisper"]:
|
||||
pass
|
||||
# user input but it is empty -> use decoder_start_token_id
|
||||
elif decoder_input_ids.size(-1)==0:
|
||||
elif decoder_input_ids.size(-1) == 0:
|
||||
decoder_input_ids = decoder_start_token_id
|
||||
# user input but doesn't start with decoder_start_token_id -> prepend decoder_start_token_id (and adjust
|
||||
# decoder_attention_mask if provided)
|
||||
|
Loading…
Reference in New Issue
Block a user