🧪 Limit max input lengths to prevent crash in CI

hezarai · Feb 5, 2024 · fae959d · fae959d
1 parent 3eb23d5
commit fae959d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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"
             }