SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix-copies

Browse Source
This commit is contained in:
Lysandre
2020-10-06 23:44:03 +02:00
parent 5982431814
commit f0d20ad328
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/transformers/utils/dummy_pt_objects.py
View File

@@ -931,6 +931,15 @@ class GPT2DoubleHeadsModel:
requires_pytorch(self) 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: class GPT2LMHeadModel:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
requires_pytorch(self) requires_pytorch(self)