fix

Signed-off-by: shunxing12345 <[email protected]>
shunxing12345 · Dec 30, 2024 · fa79011 · fa79011
1 parent cc3293d
commit fa79011
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 10 deletions.
diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
@@ -1033,7 +1033,6 @@
     _import_structure["models.gpt_sw3"].append("GPTSw3Tokenizer")
     _import_structure["models.layoutxlm"].append("LayoutXLMTokenizer")
     _import_structure["models.llama"].append("LlamaTokenizer")
-    _import_structure["models.telechat2"].append("TeleChat2Tokenizer")
     _import_structure["models.m2m_100"].append("M2M100Tokenizer")
     _import_structure["models.marian"].append("MarianTokenizer")
     _import_structure["models.mbart"].append("MBartTokenizer")
@@ -2648,9 +2647,6 @@
     _import_structure["models.telechat2"].extend(
         [
             "TeleChat2ForCausalLM",
-            "TeleChat2ForQuestionAnswering",
-            "TeleChat2ForSequenceClassification",
-            "TeleChat2ForTokenClassification",
             "TeleChat2Model",
             "TeleChat2PreTrainedModel",
         ]
@@ -4860,7 +4856,6 @@
     )
     _import_structure["models.gptj"].extend(["FlaxGPTJForCausalLM", "FlaxGPTJModel", "FlaxGPTJPreTrainedModel"])
     _import_structure["models.llama"].extend(["FlaxLlamaForCausalLM", "FlaxLlamaModel", "FlaxLlamaPreTrainedModel"])
-    _import_structure["models.telechat2"].extend(["FlaxTeleChat2ForCausalLM", "FlaxTeleChat2Model", "FlaxTeleChat2PreTrainedModel"])
     _import_structure["models.gemma"].extend(["FlaxGemmaForCausalLM", "FlaxGemmaModel", "FlaxGemmaPreTrainedModel"])
     _import_structure["models.longt5"].extend(
         [
@@ -6065,7 +6060,6 @@
         from .models.gpt_sw3 import GPTSw3Tokenizer
         from .models.layoutxlm import LayoutXLMTokenizer
         from .models.llama import LlamaTokenizer
-        from .models.telechat2 import TeleChat2Tokenizer
         from .models.m2m_100 import M2M100Tokenizer
         from .models.marian import MarianTokenizer
         from .models.mbart import MBartTokenizer
@@ -7419,9 +7413,6 @@
         )
         from .models.telechat2 import (
             TeleChat2ForCausalLM,
-            TeleChat2ForQuestionAnswering,
-            TeleChat2ForSequenceClassification,
-            TeleChat2ForTokenClassification,
             TeleChat2Model,
             TeleChat2PreTrainedModel,
         )

diff --git a/tests/models/telechat2/test_modeling_telechat2.py b/tests/models/telechat2/test_modeling_telechat2.py
@@ -296,7 +296,6 @@ def prepare_config_and_inputs_for_common(self):
 
 
 @require_torch
-# Copied from tests.models.mistral.test_modeling_mistral.MistralModelTest with Mistral->TeleChat2
 class TeleChat2ModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixin, unittest.TestCase):
     all_model_classes = (
         (