Skip to content

Commit c352640

Browse files
fix (#17310)
1 parent d9050dc commit c352640

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

src/transformers/utils/dummy_pt_objects.py

+7
Original file line numberDiff line numberDiff line change
@@ -3868,6 +3868,13 @@ def __init__(self, *args, **kwargs):
38683868
SPLINTER_PRETRAINED_MODEL_ARCHIVE_LIST = None
38693869

38703870

3871+
class SplinterForPreTraining(metaclass=DummyObject):
3872+
_backends = ["torch"]
3873+
3874+
def __init__(self, *args, **kwargs):
3875+
requires_backends(self, ["torch"])
3876+
3877+
38713878
class SplinterForQuestionAnswering(metaclass=DummyObject):
38723879
_backends = ["torch"]
38733880

0 commit comments

Comments
 (0)