riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-13 17:48:22 +06:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Proper formatting

Browse Source
This commit is contained in:
Adrian Punga 2025-01-01 14:04:50 +02:00 committed by GitHub
parent 09bbaf93e7
commit ccbae156f2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/generation/utils.py
View File

@ -699,7 +699,7 @@ class GenerationMixin:
elif self.config.model_type in ["whisper"]: elif self.config.model_type in ["whisper"]:
pass pass
# user input but it is empty -> use decoder_start_token_id # 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 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 # user input but doesn't start with decoder_start_token_id -> prepend decoder_start_token_id (and adjust
# decoder_attention_mask if provided) # decoder_attention_mask if provided)