small fix in doc

This commit is contained in:
VictorSanh
2019-06-01 16:06:50 -04:00
parent 2576a5c6db
commit 48a58646e8

View File

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