update hubconf for gpt2 torchhub compatibility

2576a5c6 · VictorSanh · a92b6dc3 · 2576a5c6
Commit 2576a5c6 authored Jun 01, 2019 by VictorSanh
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 1 deletion

hubconf.py hubconf.py +7 -1

No files found.
--- a/hubconf.py
+++ b/hubconf.py
@@ -16,4 +16,10 @@ from hubconfs.gpt_hubconf import (
    openAIGPTModel,
    openAIGPTLMHeadModel,
    openAIGPTDoubleHeadsModel
-)
\ No newline at end of file
+)
+from hubconfs.gpt2_hubconf import (
+    gpt2Tokenizer,
+    gpt2Model,
+    gpt2LMHeadModel,
+    gpt2DoubleHeadsModel
+)