Misc fixes

huggingface · Sep 6, 2024 · 3967eaa · 3967eaa
1 parent f92d17c
commit 3967eaa
Show file tree

Hide file tree

Showing 12 changed files with 32 additions and 246 deletions.
diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
@@ -3895,7 +3895,6 @@
             "TFConvBertForQuestionAnswering",
             "TFConvBertForSequenceClassification",
             "TFConvBertForTokenClassification",
-            "TFConvBertLayer",
             "TFConvBertModel",
             "TFConvBertPreTrainedModel",
         ]
@@ -4226,7 +4225,6 @@
             "TFRemBertForQuestionAnswering",
             "TFRemBertForSequenceClassification",
             "TFRemBertForTokenClassification",
-            "TFRemBertLayer",
             "TFRemBertModel",
             "TFRemBertPreTrainedModel",
         ]
@@ -4272,7 +4270,6 @@
             "TFRoFormerForQuestionAnswering",
             "TFRoFormerForSequenceClassification",
             "TFRoFormerForTokenClassification",
-            "TFRoFormerLayer",
             "TFRoFormerModel",
             "TFRoFormerPreTrainedModel",
         ]

diff --git a/src/transformers/models/align/processing_align.py b/src/transformers/models/align/processing_align.py
@@ -16,6 +16,8 @@
 Image/Text processor class for ALIGN
 """
 
+from typing import List, Union
+
 
 try:
     from typing import Unpack

diff --git a/src/transformers/models/speech_to_text/__init__.py b/src/transformers/models/speech_to_text/__init__.py
@@ -100,4 +100,5 @@
 
 else:
     import sys
+
     sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__)
-Original file line number
+Diff line change
@@ Expand Up / @@ -16,6 +16,8 @@ @@
     Image/Text processor class for ALIGN
     """
+    from typing import List, Union
     try:
         from typing import Unpack
@@ Expand Down @@
Original file line number	Diff line number	Diff line change
Expand Up		@@ -100,4 +100,5 @@

		else:
		import sys

		sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__)