🧪 Edit whisper paths in test_trainer.py

hezarai · Apr 24, 2024 · d4e4241 · d4e4241
1 parent c4c5eb5
commit d4e4241
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_trainer.py b/tests/test_trainer.py
@@ -112,12 +112,12 @@
             "num_samples": 2,
             "config": {
                 "labels_max_length": 16,
-                "tokenizer_path": "hezarai/whisper-small",
-                "feature_extractor_path": "hezarai/whisper-small"
+                "tokenizer_path": "hezarai/whisper-small-fa",
+                "feature_extractor_path": "hezarai/whisper-small-fa"
             }
         },
         "model": {
-            "path": "hezarai/whisper-small"
+            "path": "hezarai/whisper-small-fa"
         },
         "config": {
             "task": "speech_recognition",