File size: 560 Bytes
53cf398 d473317 53cf398 d473317 53cf398 d473317 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import logging
from pydantic import BaseModel
from pydantic_settings import BaseSettings
from typing import Optional, Literal
logger = logging.getLogger(__name__)
class ModelSettings(BaseSettings):
asr_model: str = "distil-whisper/distil-large-v3"
class InferenceConfig(BaseModel):
task: Literal["transcribe", "translate"] = "transcribe"
batch_size: int = 24
chunk_length_s: int = 30
sampling_rate: int = 16000
language: Optional[str] = None
model_settings = ModelSettings()
logger.info(f"asr model: {model_settings.asr_model}") |