Spaces:
Running
Running
jhj0517
commited on
Commit
·
1f2b6d3
1
Parent(s):
fefcf82
Fix compute type
Browse files
tests/test_transcription.py
CHANGED
@@ -41,6 +41,7 @@ def test_transcribe(
|
|
41 |
model_size=TEST_WHISPER_MODEL,
|
42 |
vad_filter=vad_filter,
|
43 |
is_bgm_separate=bgm_separation,
|
|
|
44 |
uvr_enable_offload=True,
|
45 |
is_diarize=diarization,
|
46 |
).as_list()
|
|
|
41 |
model_size=TEST_WHISPER_MODEL,
|
42 |
vad_filter=vad_filter,
|
43 |
is_bgm_separate=bgm_separation,
|
44 |
+
compute_type=whisper_inferencer.current_compute_type,
|
45 |
uvr_enable_offload=True,
|
46 |
is_diarize=diarization,
|
47 |
).as_list()
|