mirror of
https://github.com/QuentinFuxa/WhisperLiveKit.git
synced 2026-03-07 22:33:36 +00:00
29 lines
1.1 KiB
Python
29 lines
1.1 KiB
Python
# This code was originally in simul_whisper/transcriber/simul_whisper.py . It is adapted a lot for SimulStreaming.
|
|
|
|
from dataclasses import dataclass, field
|
|
from typing import Literal
|
|
|
|
@dataclass
|
|
class SimulWhisperConfig:
|
|
'''Options that are common for all simul policies that could be implemented in SimulWhisper.'''
|
|
model_path: str
|
|
language: str = field(default="zh")
|
|
nonspeech_prob: float = 1.0
|
|
audio_min_len: float = 1.0
|
|
decoder_type: Literal["greedy","beam"] = "greedy"
|
|
beam_size: int = 5
|
|
task: Literal["transcribe","translate"] = "transcribe"
|
|
init_prompt: str = field(default=None)
|
|
static_init_prompt: str = field(default=None)
|
|
max_context_tokens: int = field(default=None)
|
|
|
|
@dataclass
|
|
class AlignAttConfig(SimulWhisperConfig):
|
|
'''Options specific to the AlignAtt policy.'''
|
|
eval_data_path: str = "tmp"
|
|
segment_length: float = field(default=1.0, metadata = {"help": "in second"})
|
|
frame_threshold: int = 4
|
|
rewind_threshold: int = 200
|
|
audio_max_len: float = 30.0
|
|
cif_ckpt_path: str = ""
|
|
never_fire: bool = False |