jhj0517 commited on
Commit
7b06fe8
·
1 Parent(s): 38eb839

Print device info

Browse files
Files changed (1) hide show
  1. tests/test_transcription.py +5 -1
tests/test_transcription.py CHANGED
@@ -31,7 +31,11 @@ def test_transcribe(
31
  whisper_inferencer = WhisperFactory.create_whisper_inference(
32
  whisper_type=whisper_type,
33
  )
34
- print("Device : ", whisper_inferencer.device)
 
 
 
 
35
 
36
  hparams = WhisperValues(
37
  model_size=TEST_WHISPER_MODEL,
 
31
  whisper_inferencer = WhisperFactory.create_whisper_inference(
32
  whisper_type=whisper_type,
33
  )
34
+ print(
35
+ f"""Whisper Device : {whisper_inferencer.device}"""
36
+ f"""BGM Separation Device: {whisper_inferencer.music_separator.device}"""
37
+ f"""Diarization Device: {whisper_inferencer.diarizer.device}"""
38
+ )
39
 
40
  hparams = WhisperValues(
41
  model_size=TEST_WHISPER_MODEL,