fix (#17310)

c3526400 · Patrick von Platen · GitHub · d9050dc7 · c3526400
Unverified Commit c3526400 authored May 18, 2022 by Patrick von Platen Committed by GitHub May 17, 2022
Show whitespace changes
Inline Side-by-side

Showing with 7 additions and 0 deletions

src/transformers/utils/dummy_pt_objects.py src/transformers/utils/dummy_pt_objects.py +7 -0

No files found.
--- a/src/transformers/utils/dummy_pt_objects.py
+++ b/src/transformers/utils/dummy_pt_objects.py
@@ -3868,6 +3868,13 @@ class Speech2Text2PreTrainedModel(metaclass=DummyObject):
 SPLINTER_PRETRAINED_MODEL_ARCHIVE_LIST = None


+class SplinterForPreTraining(metaclass=DummyObject):
+    _backends = ["torch"]
+
+    def __init__(self, *args, **kwargs):
+        requires_backends(self, ["torch"])
+
+
 class SplinterForQuestionAnswering(metaclass=DummyObject):
    _backends = ["torch"]