simulstreaming warmup is done in whisperlivekit.simul_whisper.backend.load_model, not in warmup_online
This commit is contained in:
parent
ebaf36a8be
commit
2f6e00f512
2 changed files with 2 additions and 11 deletions
|
|
@ -4,7 +4,7 @@ try:
|
||||||
except ImportError:
|
except ImportError:
|
||||||
from .whisper_streaming_custom.whisper_online import backend_factory
|
from .whisper_streaming_custom.whisper_online import backend_factory
|
||||||
from .whisper_streaming_custom.online_asr import OnlineASRProcessor
|
from .whisper_streaming_custom.online_asr import OnlineASRProcessor
|
||||||
from whisperlivekit.warmup import warmup_asr, warmup_online
|
from whisperlivekit.warmup import warmup_asr
|
||||||
from argparse import Namespace
|
from argparse import Namespace
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
|
|
@ -155,7 +155,6 @@ def online_factory(args, asr, tokenizer, logfile=sys.stderr):
|
||||||
asr,
|
asr,
|
||||||
logfile=logfile,
|
logfile=logfile,
|
||||||
)
|
)
|
||||||
# warmup_online(online, args.warmup_file)
|
|
||||||
else:
|
else:
|
||||||
online = OnlineASRProcessor(
|
online = OnlineASRProcessor(
|
||||||
asr,
|
asr,
|
||||||
|
|
|
||||||
|
|
@ -48,12 +48,4 @@ def warmup_asr(asr, warmup_file=None, timeout=5):
|
||||||
logger.warning("Warmup file unavailable. Skipping ASR warmup.")
|
logger.warning("Warmup file unavailable. Skipping ASR warmup.")
|
||||||
return
|
return
|
||||||
asr.transcribe(audio)
|
asr.transcribe(audio)
|
||||||
logger.info("ASR model is warmed up.")
|
logger.info("ASR model is warmed up.")
|
||||||
|
|
||||||
def warmup_online(online, warmup_file=None, timeout=5):
|
|
||||||
audio = load_file(warmup_file=warmup_file, timeout=timeout)
|
|
||||||
if audio is None:
|
|
||||||
logger.warning("Warmup file unavailable. Skipping online warmup.")
|
|
||||||
return
|
|
||||||
online.warmup(audio)
|
|
||||||
logger.info("Online processor is warmed up.")
|
|
||||||
Loading…
Reference in a new issue