File size: 434 Bytes
526a387 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
"""
GSLM (Generative Spoken Language Model) - HuggingFace Integration
"""
from .config import GSLMConfig
from .modeling import GSLMForCausalLM
# Register the model with AutoModel
from transformers import AutoConfig, AutoModelForCausalLM
# Register the configuration
AutoConfig.register("gslm", GSLMConfig)
# Register the model
AutoModelForCausalLM.register(GSLMConfig, GSLMForCausalLM)
__all__ = ["GSLMConfig", "GSLMForCausalLM"] |