| 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") | |