GPT Neo few fixes (#10968)

* fix checkpoint names

* auto model

* fix doc
This commit is contained in:
Suraj Patil
2021-03-30 20:45:55 +05:30
committed by GitHub
parent 7772ddb473
commit 83d38c9ff3
7 changed files with 17 additions and 15 deletions

View File

@@ -31,8 +31,8 @@ The :obj:`generate()` method can be used to generate text using GPT Neo model.
.. code-block::
>>> from transformers import GPTNeoForCausalLM, GPT2Tokenizer
>>> model = GPTNeoForCausalLM.from_pretrained("EleutherAI/gpt_neo_xl")
>>> tokenizer = GPT2Tokenizer.from_pretrained("EleutherAI/gpt_neo_xl")
>>> model = GPTNeoForCausalLM.from_pretrained("EleutherAI/gpt-neo-1.3B")
>>> tokenizer = GPT2Tokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B")
>>> prompt = "In a shocking finding, scientists discovered a herd of unicorns living in a remote, " \
... "previously unexplored valley, in the Andes Mountains. Even more surprising to the " \