mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
whisper bugfix
This commit is contained in:
parent
d238a5ab44
commit
7f6099140a
@ -9,6 +9,7 @@ from torch import Tensor
|
||||
from torch import nn
|
||||
|
||||
import whisper
|
||||
|
||||
# import whisper_timestamped as whisper
|
||||
|
||||
from funasr.utils.load_utils import load_audio_text_image_video, extract_fbank
|
||||
@ -111,10 +112,10 @@ class WhisperWarp(nn.Module):
|
||||
|
||||
# decode the audio
|
||||
options = whisper.DecodingOptions(**kwargs.get("DecodingOptions", {}))
|
||||
|
||||
result = whisper.decode(self.model, speech, language='english')
|
||||
|
||||
result = whisper.decode(self.model, speech, options=options)
|
||||
# result = whisper.transcribe(self.model, speech)
|
||||
|
||||
|
||||
results = []
|
||||
result_i = {"key": key[0], "text": result.text}
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user