make style

huggingface · Jul 18, 2024 · 6b7b3d6 · 6b7b3d6
1 parent ed8cc34
commit 6b7b3d6
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/transformers/models/whisper/generation_whisper.py b/src/transformers/models/whisper/generation_whisper.py
@@ -722,8 +722,13 @@ def generate(
         if is_shortform:
             # add eos token:
             if generation_config.max_new_tokens is None and generation_config.max_length is None:
-
-                eos_tokens = torch.full((sequences.shape[0], 1,), generation_config.eos_token_id).to(sequences.device)
+                eos_tokens = torch.full(
+                    (
+                        sequences.shape[0],
+                        1,
+                    ),
+                    generation_config.eos_token_id,
+                ).to(sequences.device)
                 sequences = torch.cat([sequences, eos_tokens], dim=-1)
 
             if return_token_timestamps: