From d4e42416417d306c7fdb58a90afddcbcb9725b70 Mon Sep 17 00:00:00 2001 From: arxyzan Date: Thu, 25 Apr 2024 00:10:17 +0330 Subject: [PATCH] :test_tube: Edit whisper paths in `test_trainer.py` --- tests/test_trainer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_trainer.py b/tests/test_trainer.py index 34000c5e..b552a67d 100644 --- 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",