fix

Signed-off-by: shunxing12345 <[email protected]>
huggingface · Dec 30, 2024 · 775062f · 775062f
1 parent 6370c62
commit 775062f
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 13 deletions.
diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
@@ -5512,7 +5512,6 @@
     from .models.levit import LevitConfig
     from .models.lilt import LiltConfig
     from .models.llama import LlamaConfig
-    from .models.telechat2 import TeleChat2Config
     from .models.llava import (
         LlavaConfig,
         LlavaProcessor,
@@ -5799,6 +5798,7 @@
         TapasConfig,
         TapasTokenizer,
     )
+    from .models.telechat2 import TeleChat2Config
     from .models.time_series_transformer import (
         TimeSeriesTransformerConfig,
     )
@@ -7411,11 +7411,6 @@
             LlamaModel,
             LlamaPreTrainedModel,
         )
-        from .models.telechat2 import (
-            TeleChat2ForCausalLM,
-            TeleChat2Model,
-            TeleChat2PreTrainedModel,
-        )
         from .models.llava import (
             LlavaForConditionalGeneration,
             LlavaPreTrainedModel,
@@ -8131,6 +8126,11 @@
             TapasPreTrainedModel,
             load_tf_weights_in_tapas,
         )
+        from .models.telechat2 import (
+            TeleChat2ForCausalLM,
+            TeleChat2Model,
+            TeleChat2PreTrainedModel,
+        )
         from .models.time_series_transformer import (
             TimeSeriesTransformerForPrediction,
             TimeSeriesTransformerModel,
@@ -9192,11 +9192,6 @@
             FlaxLlamaModel,
             FlaxLlamaPreTrainedModel,
         )
-        from .models.telechat2 import (
-            FlaxTeleChat2ForCausalLM,
-            FlaxTeleChat2Model,
-            FlaxTeleChat2PreTrainedModel,
-        )
         from .models.longt5 import (
             FlaxLongT5ForConditionalGeneration,
             FlaxLongT5Model,

diff --git a/src/transformers/convert_slow_tokenizer.py b/src/transformers/convert_slow_tokenizer.py
@@ -1593,7 +1593,6 @@ def converted(self) -> Tokenizer:
     "SeamlessM4TTokenizer": SeamlessM4TConverter,
     "SqueezeBertTokenizer": BertConverter,
     "T5Tokenizer": T5Converter,
-    "TeleChat2Tokenizer": LlamaConverter,
     "UdopTokenizer": UdopConverter,
     "WhisperTokenizer": WhisperConverter,
     "XLMRobertaTokenizer": XLMRobertaConverter,

diff --git a/src/transformers/models/__init__.py b/src/transformers/models/__init__.py
@@ -137,7 +137,6 @@
     levit,
     lilt,
     llama,
-    telechat2,
     llava,
     llava_next,
     llava_next_video,
@@ -251,6 +250,7 @@
     t5,
     table_transformer,
     tapas,
+    telechat2,
     time_series_transformer,
     timesformer,
     timm_backbone,