Update __init__.py

__init__.py

@@ -8,6 +8,7 @@ from transformers.utils import (
 
 _import_structure = {
     "configuration_arlow_gpt": ["ARLOW_GPT_PRETRAINED_CONFIG_ARCHIVE_MAP", "ArlowGPTConfig"],
+    "tokenization_arlow_gpt": ["ArlowGPTTokenizer"],
 }
 
 try:
@@ -18,12 +19,13 @@ except OptionalDependencyNotAvailable:
 else:
     _import_structure["modeling_arlow_gpt"] = [
         "ArlowGPTModel",
-        "ArlowGPTForImageTextToText",
+        "ArlowGPTForCausalLM",
         "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():
@@ -33,7 +35,7 @@ if TYPE_CHECKING:
     else:
         from .modeling_arlow_gpt import (
             ArlowGPTModel,
-            ArlowGPTForImageTextToText,
+            ArlowGPTForCausalLM,
             ArlowGPTPreTrainedModel,
         )