admin-valentin/WhisperLiveKit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix simulstreaming imports

Browse source
This commit is contained in:
Quentin Fuxa 2025-08-19 14:43:54 +02:00
parent 90c0250ba4
commit 2699b0974c
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
whisperlivekit/core.py
View file

@ -92,7 +92,7 @@ class TranscriptionEngine:
if self.args.transcription: if self.args.transcription:
if self.args.backend == "simulstreaming": if self.args.backend == "simulstreaming":
from simul_whisper import SimulStreamingASR from whisperlivekit.simul_whisper import SimulStreamingASR
self.tokenizer = None self.tokenizer = None
simulstreaming_kwargs = {} simulstreaming_kwargs = {}
for attr in ['frame_threshold', 'beams', 'decoder_type', 'audio_max_len', 'audio_min_len', for attr in ['frame_threshold', 'beams', 'decoder_type', 'audio_max_len', 'audio_min_len',
@ -132,13 +132,17 @@ class TranscriptionEngine:
def online_factory(args, asr, tokenizer, logfile=sys.stderr): def online_factory(args, asr, tokenizer, logfile=sys.stderr):
if args.backend == "simulstreaming": if args.backend == "simulstreaming":
from simul_whisper import SimulStreamingOnlineProcessor from whisperlivekit.simul_whisper import SimulStreamingOnlineProcessor
online = SimulStreamingOnlineProcessor( online = SimulStreamingOnlineProcessor(
asr, asr,
logfile=logfile, logfile=logfile,
) )
# warmup_online(online, args.warmup_file) # warmup_online(online, args.warmup_file)
elif False: #args.vac: #vac is now handled in audio_processor elif False: #elif args.vac:
"""
WhisperStreaming backend also offer to use VAC.
but we handle it directly in the audio processor, which is more efficient
"""
online = VACOnlineASRProcessor( online = VACOnlineASRProcessor(
args.min_chunk_size, args.min_chunk_size,
asr, asr,