Update __init__.py

__init__.py

@@ -1,3 +1,4 @@
+# __init__.py
 from typing import TYPE_CHECKING
 
 from transformers.utils import (
@@ -8,7 +9,6 @@ from transformers.utils import (
 
 _import_structure = {
     "configuration_arlow_gpt": ["ARLOW_GPT_PRETRAINED_CONFIG_ARCHIVE_MAP", "ArlowGPTConfig"],
-    "tokenization_arlow_gpt": ["ArlowGPTTokenizer"],
 }
 
 try:
@@ -19,13 +19,12 @@ except OptionalDependencyNotAvailable:
 else:
     _import_structure["modeling_arlow_gpt"] = [
         "ArlowGPTModel",
-        "ArlowGPTForCausalLM",
+        "ArlowGPTForImageTextToText",
         "ArlowGPTPreTrainedModel",
     ]
 
 if TYPE_CHECKING:
     from .configuration_arlow_gpt import ARLOW_GPT_PRETRAINED_CONFIG_ARCHIVE_MAP, ArlowGPTConfig
-    from .tokenization_arlow_gpt import ArlowGPTTokenizer
 
     try:
         if not is_torch_available():
@@ -35,7 +34,7 @@ if TYPE_CHECKING:
     else:
         from .modeling_arlow_gpt import (
             ArlowGPTModel,
-            ArlowGPTForCausalLM,
+            ArlowGPTForImageTextToText,
             ArlowGPTPreTrainedModel,
         )