From 2576a5c6db0b2b97b79c1e649a2546d9ca6182bc Mon Sep 17 00:00:00 2001 From: VictorSanh Date: Sat, 1 Jun 2019 15:28:01 -0400 Subject: [PATCH] update hubconf for gpt2 torchhub compatibility --- hubconf.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/hubconf.py b/hubconf.py index ba09cbab3c..d643e57199 100644 --- 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 +)