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,
|