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

small fix in doc

Browse Source
This commit is contained in:
VictorSanh
2019-06-01 16:06:50 -04:00
parent 2576a5c6db
commit 48a58646e8
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
hubconfs/gpt2_hubconf.py
View File

@@ -91,7 +91,6 @@ def gpt2Model(*args, **kwargs):
>>> with torch.no_grad(): >>> with torch.no_grad():
hidden_states_1, past = model(tokens_tensor_1) hidden_states_1, past = model(tokens_tensor_1)
hidden_states_2, past = model(tokens_tensor_2, past=past) hidden_states_2, past = model(tokens_tensor_2, past=past)
""" """
model = GPT2Model.from_pretrained(*args, **kwargs) model = GPT2Model.from_pretrained(*args, **kwargs)
return model return model