fix(transcriber): 优化获取转录器实例的返回值,确保返回正确的 fast-whisper 实例

This commit is contained in:
SurfRid3r
2025-04-26 00:10:04 +08:00
parent 30e6a6ddd7
commit eff6e7fe75

View File

@@ -31,15 +31,15 @@ _transcribers = {
def get_whisper_transcriber(model_size="base", device="cuda"):
"""获取 Whisper 转录器实例"""
if _transcribers['fast-whisper'] is None:
if _transcribers['fast-whisper'] is None:
logger.info(f'创建 Whisper 转录器实例,参数:{model_size}, {device}')
try:
_transcribers['whisper'] = WhisperTranscriber(model_size=model_size, device=device)
_transcribers['fast-whisper'] = WhisperTranscriber(model_size=model_size, device=device)
logger.info('Whisper 转录器创建成功')
except Exception as e:
logger.error(f"Whisper 转录器创建失败: {e}")
raise
return _transcribers['whisper']
return _transcribers['fast-whisper']
def get_bcut_transcriber():
"""获取 Bcut 转录器实例"""