| # __init__.py | |
| from .configuration_i3 import I3Config | |
| from .modeling_i3 import I3ForCausalLM | |
| from .tokenization_i3 import I3Tokenizer | |
| # Auto class registration | |
| from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer | |
| AutoConfig.register("i3", I3Config) | |
| AutoModelForCausalLM.register(I3Config, I3ForCausalLM) | |
| AutoTokenizer.register(I3Config, I3Tokenizer) |