whispering/whisper_streaming/schema.py
2022-09-23 22:39:27 +09:00

33 lines
833 B
Python

#!/usr/bin/env python3
from typing import List, Optional, Tuple
from pydantic import BaseModel
class WhisperConfig(BaseModel):
model_name: str
device: str
language: str
fp16: bool = True
temperatures: Tuple[float, ...] = (0.0, 0.2, 0.4, 0.6, 0.8, 1.0)
compression_ratio_threshold: Optional[float] = 2.4
logprob_threshold: Optional[float] = -1.0
no_captions_threshold: Optional[float] = 0.6
best_of: int = 5
beam_size: Optional[int] = None
no_speech_threshold: Optional[float] = 0.6
logprob_threshold: Optional[float] = -1.0
compression_ratio_threshold: Optional[float] = 2.4
class ParsedChunk(BaseModel):
start: float
end: float
text: str
tokens: List[int]
temperature: float
avg_logprob: float
compression_ratio: float
no_speech_prob: float