admin/scribe
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

set test whisper model to tiny

Browse Source
This commit is contained in:
Schmieder, Jacob
2024-09-10 15:22:18 +00:00
parent 929f916077
commit ab7b43ac48
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/test_transcriber.py
+3 -3
View File
@@ -31,12 +31,12 @@ def test_transcriber(mock_load_model, audio_file, expected_transcription):
@pytest.fixture @pytest.fixture
def whisper_instance(): def whisper_instance():
return load_transcriber('medium', whisper_type='whisper') return load_transcriber('tiny', whisper_type='whisper')
@pytest.fixture @pytest.fixture
def faster_whisper_instance(): def faster_whisper_instance():
return load_transcriber('medium', whisper_type='faster-whisper') return load_transcriber('tiny', whisper_type='faster-whisper')
def test_whisper_base_initialization(whisper_instance): def test_whisper_base_initialization(whisper_instance):
@@ -57,7 +57,7 @@ def test_faster_whisper_transcriber_initialization(faster_whisper_instance):
def test_wrong_transcriber_initialization(): def test_wrong_transcriber_initialization():
with pytest.raises(ValueError): with pytest.raises(ValueError):
load_transcriber('medium', whisper_type='wrong_whisper') load_transcriber('tiny', whisper_type='wrong_whisper')
def test_get_whisper_kwargs(): def test_get_whisper_kwargs():