Fix imcomplete sentence

huggingface/transformers#16622
ayaka14732 · Apr 6, 2022 · b6b51f9 · b6b51f9
1 parent ee63dc7
commit b6b51f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nmt_generate.py b/nmt_generate.py
@@ -71,7 +71,7 @@ def fwd_encode(params: dict, src: np.ndarray, mask_enc: np.ndarray) -> np.ndarra
 mask_enc = np.einsum('bi,bj->bij', mask_enc_1d, mask_enc_1d)[:, None]
 
 encoder_last_hidden_output = fwd_encode(params, src, mask_enc)
-generate_ids = generator.generate(encoder_last_hidden_output, mask_enc_1d, num_beams=5)
+generate_ids = generator.generate(encoder_last_hidden_output, mask_enc_1d, num_beams=5, max_length=100000)
 decoded_sentences = tokenizer_en.batch_decode(generate_ids, skip_special_tokens=True)
 
 for translated_sentence in decoded_sentences: