Fix imports, add model docs

huggingface · Oct 13, 2022 · ab774a6 · ab774a6
1 parent c2a67a8
commit ab774a6
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/docs/source/en/model_doc/esm.mdx b/docs/source/en/model_doc/esm.mdx
@@ -107,3 +107,23 @@ and [Matt](https://huggingface.co/Rocketknight1).
 
 [[autodoc]] EsmForTokenClassification
     - forward
+
+## TFEsmModel
+
+[[autodoc]] TFEsmModel
+    - call
+
+## TFEsmForMaskedLM
+
+[[autodoc]] TFEsmForMaskedLM
+    - call
+
+## TFEsmForSequenceClassification
+
+[[autodoc]] TFEsmForSequenceClassification
+    - call
+
+## TFEsmForTokenClassification
+
+[[autodoc]] TFEsmForTokenClassification
+    - call
diff --git a/src/transformers/models/esm/__init__.py b/src/transformers/models/esm/__init__.py
@@ -47,7 +47,7 @@
     pass
 else:
     _import_structure["modeling_tf_esm"] = [
-        "ESM_PRETRAINED_MODEL_ARCHIVE_LIST",
+        "TF_ESM_PRETRAINED_MODEL_ARCHIVE_LIST",
         "TFEsmForMaskedLM",
         "TFEsmForSequenceClassification",
         "TFEsmForTokenClassification",
@@ -82,7 +82,7 @@
         pass
     else:
         from .modeling_tf_esm import (
-            ESM_PRETRAINED_MODEL_ARCHIVE_LIST,
+            TF_ESM_PRETRAINED_MODEL_ARCHIVE_LIST,
             TFEsmForMaskedLM,
             TFEsmForSequenceClassification,
             TFEsmForTokenClassification,