From f5aeae21bd6c76a3a34a32c2ba099b5f048751ef Mon Sep 17 00:00:00 2001 From: Alishahryar1 Date: Mon, 16 Feb 2026 20:32:24 -0800 Subject: [PATCH] fixed weight warning --- messaging/transcription.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/messaging/transcription.py b/messaging/transcription.py index 03abc17f..e42dde41 100644 --- a/messaging/transcription.py +++ b/messaging/transcription.py @@ -46,7 +46,13 @@ def _get_local_model(whisper_model: str, device: str) -> _WhisperModelLike: try: _model_cache[cache_key] = WhisperModel(whisper_model, device="cuda") except RuntimeError: - _model_cache[cache_key] = WhisperModel(whisper_model, device="cpu") + _model_cache[cache_key] = WhisperModel( + whisper_model, device="cpu", compute_type="float32" + ) + elif resolved == "cpu": + _model_cache[cache_key] = WhisperModel( + whisper_model, device="cpu", compute_type="float32" + ) else: _model_cache[cache_key] = WhisperModel(whisper_model, device=resolved) except ImportError as e: