update hubconf for gpt2 torchhub compatibility

2019-06-01 15:28:01 -04:00
parent a92b6dc3c1
commit 2576a5c6db
1 changed files with 7 additions and 1 deletions
--- a/hubconf.py
+++ b/hubconf.py
@@ -17,3 +17,9 @@ from hubconfs.gpt_hubconf import (
    openAIGPTLMHeadModel,
    openAIGPTDoubleHeadsModel
 )
+from hubconfs.gpt2_hubconf import (
+    gpt2Tokenizer,
+    gpt2Model,
+    gpt2LMHeadModel,
+    gpt2DoubleHeadsModel
+)