Merge pull request #691 from vanche/master

import class "GPT2MultipleChoiceHead"

Merge pull request #691 from vanche/master
import class "GPT2MultipleChoiceHead"
cc6cd430 · Thomas Wolf · GitHub · b3f9e945 · 8289646d · cc6cd430
Unverified Commit cc6cd430 authored Jun 15, 2019 by Thomas Wolf Committed by GitHub Jun 15, 2019
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

pytorch_pretrained_bert/__init__.py pytorch_pretrained_bert/__init__.py +1 -1

No files found.
--- a/pytorch_pretrained_bert/__init__.py
+++ b/pytorch_pretrained_bert/__init__.py
@@ -15,7 +15,7 @@ from .modeling_openai import (OpenAIGPTConfig, OpenAIGPTModel,
 from .modeling_transfo_xl import (TransfoXLConfig, TransfoXLModel, TransfoXLLMHeadModel,
                                  load_tf_weights_in_transfo_xl)
 from .modeling_gpt2 import (GPT2Config, GPT2Model,
-                            GPT2LMHeadModel, GPT2DoubleHeadsModel,
+                            GPT2LMHeadModel, GPT2DoubleHeadsModel, GPT2MultipleChoiceHead,
                            load_tf_weights_in_gpt2)

 from .optimization import BertAdam