whispering/whisper_streaming/schema.py

33 lines
811 B
Python
Raw Normal View History

2022-09-23 10:20:11 +00:00
#!/usr/bin/env python3
2022-09-23 11:03:00 +00:00
from typing import List, Optional, Tuple
2022-09-23 10:20:11 +00:00
from pydantic import BaseModel
class WhisperConfig(BaseModel):
model_name: str
device: str
language: str
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
2022-09-23 11:03:00 +00:00
class ParsedChunk(BaseModel):
start: float
end: float
text: str
tokens: List[int]
temperature: float
avg_logprob: float
compression_ratio: float
no_speech_prob: float