GithubBackup/BiliNote
1
0
Fork 0
mirror of https://github.com/JefferyHcool/BiliNote.git synced 2026-09-05 15:46:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(transcriber): 更新转录器支持和模型下载逻辑

Browse Source 
- 修改 NoteGenerator 类以支持动态选择转录器类型。
- 更新 MLXWhisperTranscriber 类，添加模型下载逻辑，确保模型存在时自动下载。
- 在 transcriber_provider.py 中优化 MLX Whisper 的环境变量处理，确保在不可用时回退到 fast-whisper。
This commit is contained in:
SurfRid3r
2025-04-20 22:50:07 +08:00
parent 369de19572
commit a567788448
3 changed files with 18 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/app/transcriber/transcriber_provider.py
+1 -1
View File
@@ -98,9 +98,9 @@ def get_transcriber(transcriber_type="fast-whisper", model_size="base", device="
whisper_model_size = os.environ.get("WHISPER_MODEL_SIZE",model_size)
return get_whisper_transcriber(whisper_model_size, device=device)
elif transcriber_type == "mlx-whisper":
whisper_model_size = os.environ.get("WHISPER_MODEL_SIZE",model_size)
if not MLX_WHISPER_AVAILABLE:
logger.warning("MLX Whisper 不可用，回退到 fast-whisper")
whisper_model_size = os.environ.get("WHISPER_MODEL_SIZE",model_size)
return get_whisper_transcriber(whisper_model_size, device=device)
return get_mlx_whisper_transcriber(whisper_model_size)
elif transcriber_type == "bcut":