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()
|