update hubconf for gpt2 torchhub compatibility

xijiu9 · Jun 1, 2019 · 2576a5c · 2576a5c
1 parent a92b6dc
commit 2576a5c
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/hubconf.py b/hubconf.py
@@ -16,4 +16,10 @@
     openAIGPTModel,
     openAIGPTLMHeadModel,
     openAIGPTDoubleHeadsModel
-)
+)
+from hubconfs.gpt2_hubconf import (
+    gpt2Tokenizer,
+    gpt2Model,
+    gpt2LMHeadModel,
+    gpt2DoubleHeadsModel
+)