From f0d20ad328b3baf7eca36d56e9404a694f538f66 Mon Sep 17 00:00:00 2001 From: Lysandre Date: Tue, 6 Oct 2020 23:44:03 +0200 Subject: [PATCH] Fix-copies --- src/transformers/utils/dummy_pt_objects.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/src/transformers/utils/dummy_pt_objects.py b/src/transformers/utils/dummy_pt_objects.py index ab854e5e74..7ac273b59d 100644 --- 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)