Skip to content

Commit 35de885

Browse files
committed
Update beam_size name
1 parent 373776b commit 35de885

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

beginner_source/t5_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -406,7 +406,7 @@ def process_labels(labels, x):
406406
beam_size = 4
407407

408408
model_input = transform(input_text)
409-
model_output = sequence_generator.generate(model_input, eos_idx=eos_idx, beam_size=beam_size)
409+
model_output = sequence_generator.generate(model_input, eos_idx=eos_idx, num_beams=beam_size)
410410
output_text = transform.decode(model_output.tolist())
411411

412412
for i in range(multi_batch_size):

0 commit comments

Comments
 (0)