GithubBackup/BiliNote
1
0
Fork 0
mirror of https://github.com/JefferyHcool/BiliNote.git synced 2026-07-12 16:11:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(transcriber): 修复转义器初始化异常的问题，fixes #36

Browse Source
This commit is contained in:
思诺特
2025-04-25 11:57:53 +08:00
parent 4012a45c93
commit 84da8ba7c2
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
backend/app/transcriber/transcriber_provider.py
View File

@@ -23,15 +23,15 @@ logger.info('初始化转录服务提供器')
# 维护各种转录器的单例实例 # 维护各种转录器的单例实例
_transcribers = { _transcribers = {
'whisper': None,
'bcut': None, 'bcut': None,
'kuaishou': None, 'kuaishou': None,
'mlx-whisper': None 'mlx-whisper': None,
'fast-whisper':None
} }
def get_whisper_transcriber(model_size="base", device="cuda"): def get_whisper_transcriber(model_size="base", device="cuda"):
"""获取 Whisper 转录器实例""" """获取 Whisper 转录器实例"""
if _transcribers['whisper'] is None: if _transcribers['fast-whisper'] is None:
logger.info(f'创建 Whisper 转录器实例，参数：{model_size}, {device}') logger.info(f'创建 Whisper 转录器实例，参数：{model_size}, {device}')
try: try:
_transcribers['whisper'] = WhisperTranscriber(model_size=model_size, device=device) _transcribers['whisper'] = WhisperTranscriber(model_size=model_size, device=device)