Skip to content

Commit fb2ee61

Browse files
committed
add dummies and we should be good to go!z
1 parent bf2bd24 commit fb2ee61

File tree

1 file changed

+35
-0
lines changed

1 file changed

+35
-0
lines changed

src/transformers/utils/dummy_pt_objects.py

+35
Original file line numberDiff line numberDiff line change
@@ -4981,6 +4981,41 @@ def __init__(self, *args, **kwargs):
49814981
requires_backends(self, ["torch"])
49824982

49834983

4984+
class HeliumForCausalLM(metaclass=DummyObject):
4985+
_backends = ["torch"]
4986+
4987+
def __init__(self, *args, **kwargs):
4988+
requires_backends(self, ["torch"])
4989+
4990+
4991+
class HeliumForSequenceClassification(metaclass=DummyObject):
4992+
_backends = ["torch"]
4993+
4994+
def __init__(self, *args, **kwargs):
4995+
requires_backends(self, ["torch"])
4996+
4997+
4998+
class HeliumForTokenClassification(metaclass=DummyObject):
4999+
_backends = ["torch"]
5000+
5001+
def __init__(self, *args, **kwargs):
5002+
requires_backends(self, ["torch"])
5003+
5004+
5005+
class HeliumModel(metaclass=DummyObject):
5006+
_backends = ["torch"]
5007+
5008+
def __init__(self, *args, **kwargs):
5009+
requires_backends(self, ["torch"])
5010+
5011+
5012+
class HeliumPreTrainedModel(metaclass=DummyObject):
5013+
_backends = ["torch"]
5014+
5015+
def __init__(self, *args, **kwargs):
5016+
requires_backends(self, ["torch"])
5017+
5018+
49845019
class HieraBackbone(metaclass=DummyObject):
49855020
_backends = ["torch"]
49865021

0 commit comments

Comments
 (0)