riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 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
Show Stats 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,
openAIGPTLMHeadModel,
openAIGPTDoubleHeadsModel
)
)
from hubconfs.gpt2_hubconf import (
gpt2Tokenizer,
gpt2Model,
gpt2LMHeadModel,
gpt2DoubleHeadsModel
)