File size: 302 Bytes
0acbed8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 | """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"]
|