Update pipeline/asr_diarization.py
Browse files
pipeline/asr_diarization.py
CHANGED
|
@@ -7,7 +7,8 @@ _metadata = None
|
|
| 7 |
def load_asr(device="cpu", model_size="medium"):
|
| 8 |
global _asr_model
|
| 9 |
if _asr_model is None:
|
| 10 |
-
|
|
|
|
| 11 |
return _asr_model
|
| 12 |
|
| 13 |
def transcribe_and_align(audio_path: str, device="cpu"):
|
|
|
|
| 7 |
def load_asr(device="cpu", model_size="medium"):
|
| 8 |
global _asr_model
|
| 9 |
if _asr_model is None:
|
| 10 |
+
# force float32 so it doesn’t try to use unsupported float16 on CPU / limited GPU
|
| 11 |
+
_asr_model = whisperx.load_model(model_size, device=device, compute_type="float32")
|
| 12 |
return _asr_model
|
| 13 |
|
| 14 |
def transcribe_and_align(audio_path: str, device="cpu"):
|