Update main.py

Wenyueh · Sep 25, 2023 · 9b6dd79 · 9b6dd79
1 parent 786adde
commit 9b6dd79
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -56,7 +56,7 @@ def predict_outputs(args, batch, model, tokenizer, prefix_allowed_tokens, k=20):
             prediction = model.module.generate(
                 input_ids=input_ids,
                 attention_mask=attn,
-                max_length=10,
+                max_length=20,
                 prefix_allowed_tokens_fn=prefix_allowed_tokens,
                 num_beams=20,
                 num_return_sequences=20,