jhj0517 commited on
Commit
1f2b6d3
·
1 Parent(s): fefcf82

Fix compute type

Browse files
Files changed (1) hide show
  1. tests/test_transcription.py +1 -0
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()