awesome-bookmarks/free-claude-code
3
0
Fork 0
mirror of https://github.com/Alishahryar1/free-claude-code.git synced 2026-07-03 14:05:26 +02:00
Code Issues Packages Projects Releases Wiki Activity

fixed weight warning

Browse Source
This commit is contained in:
Alishahryar1
2026-02-16 20:32:24 -08:00
parent 2d68906f65
commit f5aeae21bd
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
messaging/transcription.py
+7 -1
View File
@@ -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: