Skip to content

GitLab

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

small fix in doc

parent 2576a5c6
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
hubconfs/gpt2_hubconf.py
View file @ 48a58646
...@@ -91,7 +91,6 @@ def gpt2Model(*args, **kwargs): ...@@ -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
......
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