Commit 48a58646 authored by VictorSanh's avatar VictorSanh
Browse files

small fix in doc

parent 2576a5c6
......@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment