__init__.py

# __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)