fix: tests for model trainer command

KevKibe · Nov 26, 2024 · be97cda · be97cda
1 parent 2864bd9
commit be97cda
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/unit_test_model_trainer_command.yaml b/.github/workflows/unit_test_model_trainer_command.yaml
@@ -42,4 +42,4 @@ jobs:
 
       - name: Run command test
         run: |
-          cd src && python3 -m training.main --huggingface_token ${{ secrets.HF_TOKEN }}  --dataset_name mozilla-foundation/common_voice_16_1 --train_num_samples 10 --test_num_samples 10 --language_abbr sw --model_id openai/whisper-tiny --processing_task transcribe --report_to None --wandb_api_key "" --max_steps 15 --train_batch_size 8 --eval_batch_size 8 --save_eval_logging_steps 10
+          cd src && python3 -m training.main --huggingface_token ${{ secrets.HF_TOKEN }}  --dataset_name mozilla-foundation/common_voice_16_1 --train_num_samples 10 --test_num_samples 10 --language_abbr af --model_id openai/whisper-tiny --processing_task transcribe --report_to None --wandb_api_key "" --max_steps 15 --train_batch_size 8 --eval_batch_size 8 --save_eval_logging_steps 10