Gpt1 for sequence classification (#7683)

* Add Documentation for GPT-1 Classification

* Add GPT-1 with Classification head

* Add tests for GPT-1 Classification

* Add GPT-1 For Classification to auto models

* Remove authorized missing keys, change checkpoint to openai-gpt
This commit is contained in:
Felipe Curti
2020-10-13 06:06:15 -03:00
committed by GitHub
parent f34b4cd1bd
commit dcba9ee03b
6 changed files with 153 additions and 4 deletions

View File

@@ -1256,6 +1256,15 @@ class OpenAIGPTDoubleHeadsModel:
requires_pytorch(self)
class OpenAIGPTForSequenceClassification:
def __init__(self, *args, **kwargs):
requires_pytorch(self)
@classmethod
def from_pretrained(self, *args, **kwargs):
requires_pytorch(self)
class OpenAIGPTLMHeadModel:
def __init__(self, *args, **kwargs):
requires_pytorch(self)