Spaces:
Running
Running
jhj0517
commited on
Commit
·
7b06fe8
1
Parent(s):
38eb839
Print device info
Browse files
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(
|
|
|
|
|
|
|
|
|
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,
|