Create __init__.py

__init__.py

@@ -0,0 +1,14 @@
+# __init__.py
+
+# Expose the main components of the i3 model package
+from .i3_model import i3Model, ChunkTokenizer
+from .modeling_i3 import I3ForCausalLM, I3Config
+from .tokenizer_i3 import I3Tokenizer
+
+__all__ = [
+    "i3Model",
+    "ChunkTokenizer",
+    "I3ForCausalLM",
+    "I3Config",
+    "I3Tokenizer",
+]