Fix-copies

2020-10-06 23:44:03 +02:00
parent 5982431814
commit f0d20ad328
1 changed files with 9 additions and 0 deletions
--- a/src/transformers/utils/dummy_pt_objects.py
+++ b/src/transformers/utils/dummy_pt_objects.py
@@ -931,6 +931,15 @@ class GPT2DoubleHeadsModel:
        requires_pytorch(self)


+class GPT2ForSequenceClassification:
+    def __init__(self, *args, **kwargs):
+        requires_pytorch(self)
+
+    @classmethod
+    def from_pretrained(self, *args, **kwargs):
+        requires_pytorch(self)
+
+
 class GPT2LMHeadModel:
    def __init__(self, *args, **kwargs):
        requires_pytorch(self)