Update __init__.py

__init__.py

@@ -0,0 +1,11 @@
+# __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)