LFG-3 / configuration_lfg3.py
glenn2's picture
Add LFG-3 code and configuration
7e5c0ea verified
Raw
History Blame Contribute Delete
603 Bytes
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