"""Hugging Face AutoModel bridge for TimeBraid."""

from ._timebraid_hf_runtime import TimeBraidRuntimeLoader


class TimeBraid(TimeBraidRuntimeLoader):
    """Return the native model while preserving its mixed generation methods."""

    _auto_class = "AutoModelForCausalLM"


__all__ = ["TimeBraid"]