mirror of
https://github.com/JefferyHcool/BiliNote.git
synced 2026-09-09 01:36:41 +08:00
Merge pull request #40 from SurfRid3r/master
fix(transcriber): 优化获取转录器实例的返回值,确保返回正确的 fast-whisper 实例
This commit is contained in:
@@ -34,12 +34,12 @@ def get_whisper_transcriber(model_size="base", device="cuda"):
|
|||||||
if _transcribers['fast-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['fast-whisper'] = WhisperTranscriber(model_size=model_size, device=device)
|
||||||
logger.info('Whisper 转录器创建成功')
|
logger.info('Whisper 转录器创建成功')
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(f"Whisper 转录器创建失败: {e}")
|
logger.error(f"Whisper 转录器创建失败: {e}")
|
||||||
raise
|
raise
|
||||||
return _transcribers['whisper']
|
return _transcribers['fast-whisper']
|
||||||
|
|
||||||
def get_bcut_transcriber():
|
def get_bcut_transcriber():
|
||||||
"""获取 Bcut 转录器实例"""
|
"""获取 Bcut 转录器实例"""
|
||||||
|
|||||||
Reference in New Issue
Block a user