From fae959d922d3b82b6d312b5f160b489301402451 Mon Sep 17 00:00:00 2001
From: arxyzan <arxyzan@gmail.com>
Date: Mon, 5 Feb 2024 17:53:19 +0330
Subject: [PATCH] :test_tube: Limit max input lengths to prevent crash in CI

---
 tests/test_trainer.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/test_trainer.py b/tests/test_trainer.py
index 6ee7fc84..bfe3b7b0 100644
--- a/tests/test_trainer.py
+++ b/tests/test_trainer.py
@@ -52,6 +52,7 @@
             "path": "hezarai/xlsum-fa",
             "config": {
                 "tokenizer_path": "hezarai/t5-base-fa",
+                "max_length": 32
             }
         },
         "model": {
@@ -98,7 +99,7 @@
         "dataset": {
             "path": "hezarai/common-voice-13-fa",
             "config": {
-                "labels_max_length": 64,
+                "labels_max_length": 32,
                 "tokenizer_path": "hezarai/whisper-small",
                 "feature_extractor_path": "hezarai/whisper-small"
             }