Proper formatting

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

View File

@ -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)