from .configuration_bitnet import BitNetConfig
from .modeling_bitnet import BitNetModel
from transformers import AutoConfig, AutoModel

# Registrar clases en AutoConfig y AutoModel
AutoConfig.register("bitnet", BitNetConfig)
AutoModel.register(BitNetConfig, BitNetModel)

# Registrar para auto_class en el Hub
BitNetConfig.register_for_auto_class()
BitNetModel.register_for_auto_class("AutoModel")