update init

huggingface · Aug 27, 2022 · ad8e085 · ad8e085
1 parent 2c544f5
commit ad8e085
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 12 deletions.
diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
@@ -907,7 +907,6 @@
             "ErnieLMHeadModel",
             "ErnieModel",
             "ErniePreTrainedModel",
-            "load_tf_weights_in_ernie",
         ]
     )
     _import_structure["models.bert_generation"].extend(
@@ -3633,7 +3632,6 @@
             ErnieLMHeadModel,
             ErnieModel,
             ErniePreTrainedModel,
-            load_tf_weights_in_ernie,
         )
         from .models.bert_generation import (
             BertGenerationDecoder,

diff --git a/src/transformers/models/ernie/__init__.py b/src/transformers/models/ernie/__init__.py
@@ -49,7 +49,6 @@
         "ErnieLMHeadModel",
         "ErnieModel",
         "ErniePreTrainedModel",
-        "load_tf_weights_in_ernie",
     ]
 
 try:
@@ -82,17 +81,8 @@
             ErnieLMHeadModel,
             ErnieModel,
             ErniePreTrainedModel,
-            load_tf_weights_in_ernie,
         )
 
-    try:
-        if not is_tensorflow_text_available():
-            raise OptionalDependencyNotAvailable()
-    except OptionalDependencyNotAvailable:
-        pass
-    else:
-
 else:
     import sys
-
     sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__)