SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

update hubconf for gpt2 torchhub compatibility

Browse Source
This commit is contained in:
VictorSanh
2019-06-01 15:28:01 -04:00
parent a92b6dc3c1
commit 2576a5c6db
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
hubconf.py
View File

@@ -16,4 +16,10 @@ from hubconfs.gpt_hubconf import (
openAIGPTModel, openAIGPTModel,
openAIGPTLMHeadModel, openAIGPTLMHeadModel,
openAIGPTDoubleHeadsModel openAIGPTDoubleHeadsModel
) )
from hubconfs.gpt2_hubconf import (
gpt2Tokenizer,
gpt2Model,
gpt2LMHeadModel,
gpt2DoubleHeadsModel
)