@@ -1555,7 +1555,6 @@ def greedy_search(
15551555 used for text-decoder, text-to-text, speech-to-text, and vision-to-text models.
15561556
15571557 Parameters:
1558-
15591558 input_ids (`torch.LongTensor` of shape `(batch_size, sequence_length)`):
15601559 The sequence used as a prompt for the generation.
15611560 logits_processor (`LogitsProcessorList`, *optional*):
@@ -1789,7 +1788,6 @@ def sample(
17891788 can be used for text-decoder, text-to-text, speech-to-text, and vision-to-text models.
17901789
17911790 Parameters:
1792-
17931791 input_ids (`torch.LongTensor` of shape `(batch_size, sequence_length)`):
17941792 The sequence used as a prompt for the generation.
17951793 logits_processor (`LogitsProcessorList`, *optional*):
@@ -2046,7 +2044,6 @@ def beam_search(
20462044 can be used for text-decoder, text-to-text, speech-to-text, and vision-to-text models.
20472045
20482046 Parameters:
2049-
20502047 input_ids (`torch.LongTensor` of shape `(batch_size, sequence_length)`):
20512048 The sequence used as a prompt for the generation.
20522049 beam_scorer (`BeamScorer`):
@@ -2355,7 +2352,6 @@ def beam_sample(
23552352 sampling** and can be used for text-decoder, text-to-text, speech-to-text, and vision-to-text models.
23562353
23572354 Parameters:
2358-
23592355 input_ids (`torch.LongTensor` of shape `(batch_size, sequence_length)`):
23602356 The sequence used as a prompt for the generation.
23612357 beam_scorer (`BeamScorer`):
@@ -2672,7 +2668,6 @@ def group_beam_search(
26722668 decoding** and can be used for text-decoder, text-to-text, speech-to-text, and vision-to-text models.
26732669
26742670 Parameters:
2675-
26762671 input_ids (`torch.LongTensor` of shape `(batch_size, sequence_length)`):
26772672 The sequence used as a prompt for the generation.
26782673 beam_scorer (`BeamScorer`):
0 commit comments