diff --git a/PyTorch/built-in/nlp/mBART_ID2372_for_PyTorch/generate_on_en_de.sh b/PyTorch/built-in/nlp/mBART_ID2372_for_PyTorch/generate_on_en_de.sh
index 1a202ccee400a1cbac57920634cbebc51349d5fd..7e13b087392c54d9996f155cc1f409c63298ba0d 100644
--- a/PyTorch/built-in/nlp/mBART_ID2372_for_PyTorch/generate_on_en_de.sh
+++ b/PyTorch/built-in/nlp/mBART_ID2372_for_PyTorch/generate_on_en_de.sh
@@ -8,7 +8,7 @@ HYP=hyp
 REF=ref
 
 fairseq-generate $DATA_PATH \
-  --fp16 --path $model_dir --max-tokens 4096 \
+  --fp16 --path $model_dir --max-tokens 1024 \
   --task translation_from_pretrained_bart \
   --gen-subset test \
   -t de_DE -s en_XX \
diff --git a/PyTorch/built-in/nlp/mBART_ID2372_for_PyTorch/generate_on_en_ro.sh b/PyTorch/built-in/nlp/mBART_ID2372_for_PyTorch/generate_on_en_ro.sh
index 66ea0d5ece72d47c0037fca9de111823868a8232..18e527b1086180eb5d05a78bcbc9353017d2a811 100644
--- a/PyTorch/built-in/nlp/mBART_ID2372_for_PyTorch/generate_on_en_ro.sh
+++ b/PyTorch/built-in/nlp/mBART_ID2372_for_PyTorch/generate_on_en_ro.sh
@@ -16,7 +16,7 @@ HYP=hyp
 REF=ref
 
 fairseq-generate $DATA_PATH \
-  --fp16 --path $model_dir --max-tokens 4096 \
+  --fp16 --path $model_dir --max-tokens 1024 \
   --task translation_from_pretrained_bart \
   --gen-subset test \
   -t ro_RO -s en_XX \