from __future__ import annotations

from transformers import Gemma4Config


class LFG3Config(Gemma4Config):
    model_type = "gemma4"

    def __init__(
        self,
        parakeet_name: str = "nvidia/parakeet-tdt-0.6b-v3",
        projector_hidden: int = 4096,
        projector_file: str = "projector_final.pt",
        audio_frame_rate: float = 12.5,
        **kwargs,
    ):
        super().__init__(**kwargs)
        self.parakeet_name = parakeet_name
        self.projector_hidden = projector_hidden
        self.projector_file = projector_file
        self.audio_frame_rate = audio_frame_rate