Update src/transformers/generation/utils.py

gante · zucchini-nlp · gante · commit dd5bf8ccdc30 · 2024-03-29T12:23:34.000Z
Co-authored-by: Raushan Turganbay &lt;raushan.turganbay@alumni.nu.edu.kz&gt;
diff --git a/src/transformers/generation/utils.py b/src/transformers/generation/utils.py
@@ -541,7 +541,7 @@ def _prepare_decoder_input_ids_for_generation(
         if decoder_start_token_id.ndim == 1:
             if decoder_start_token_id.shape[0] != batch_size:
                 raise ValueError(
-                    f"`decoder_start_token_id` expcted to have length {batch_size} but got {decoder_start_token_id.shape[0]}"
+                    f"`decoder_start_token_id` expected to have length {batch_size} but got {decoder_start_token_id.shape[0]}"
                 )
             decoder_start_token_id = decoder_start_token_id.view(-1, 1)
         else:

Original file line number	Diff line number	Diff line change
`@@ -541,7 +541,7 @@ def _prepare_decoder_input_ids_for_generation(`
`541`	`541`	`if decoder_start_token_id.ndim == 1:`
`542`	`542`	`if decoder_start_token_id.shape[0] != batch_size:`
`543`	`543`	`raise ValueError(`
`544`		- f"`decoder_start_token_id` expcted to have length {batch_size} but got {decoder_start_token_id.shape[0]}"
	`544`	+ f"`decoder_start_token_id` expected to have length {batch_size} but got {decoder_start_token_id.shape[0]}"
`545`	`545`	`)`
`546`	`546`	`decoder_start_token_id = decoder_start_token_id.view(-1, 1)`
`547`	`547`	`else:`