From d3d3025b7d502cf951327cbcc782aeab7d4897b6 Mon Sep 17 00:00:00 2001
From: ydshieh <ydshieh@users.noreply.github.com>
Date: Wed, 17 Apr 2024 05:48:04 +0200
Subject: [PATCH] fix

---
 examples/pytorch/test_pytorch_examples.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/examples/pytorch/test_pytorch_examples.py b/examples/pytorch/test_pytorch_examples.py
index 1d4f8db9259087..9c86347f98c243 100644
--- a/examples/pytorch/test_pytorch_examples.py
+++ b/examples/pytorch/test_pytorch_examples.py
@@ -372,6 +372,7 @@ def test_run_translation(self):
             --predict_with_generate
             --source_lang en_XX
             --target_lang ro_RO
+            --max_source_length 512
         """.split()
 
         with patch.object(sys, "argv", testargs):