mrs83's picture
Upload 10 files
1ad193e verified
Raw
History Blame Contribute Delete
1.25 kB
"""
echo_hybrid/__init__.py
────────────────────────────────────────────────────────────────────────────
Package init: registers echo_hybrid classes with HuggingFace AutoClass so
that AutoConfig.from_pretrained() and AutoModelForCausalLM.from_pretrained()
work transparently without trust_remote_code=True.
Usage
─────
import echo_hybrid # must be imported before any AutoClass call
model = AutoModelForCausalLM.from_pretrained(
"models/Echo-Hybrid-0.5B-Base",
device_map="auto",
)
"""
from transformers import AutoConfig, AutoModelForCausalLM
from .configuration_hybrid import HybridEchoConfig
from .dsrn_memory_block import DSRNMemoryInjector
from .modeling_hybrid import HybridEchoCache, HybridEchoForCausalLM, HybridEchoModel
# Register with HuggingFace so AutoClass routing works
AutoConfig.register("echo_hybrid", HybridEchoConfig)
AutoModelForCausalLM.register(HybridEchoConfig, HybridEchoForCausalLM)
__all__ = [
"HybridEchoConfig",
"DSRNMemoryInjector",
"HybridEchoModel",
"HybridEchoForCausalLM",
"HybridEchoCache",
]