EleutherAI
/
gpt-j-6B
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README.md

This commit is contained in:
Stella Biderman 2021-08-06 19:54:23 +00:00 committed by huggingface-web
parent 8a44d5a1b4
commit 7cb542864a
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
README.md
View File

@ -52,12 +52,14 @@ GPT-J learns an inner representation of the English language that can be used to
### How to use
You can use this model directly with a pipeline for text generation. This example generates a different sequence each time it's run:
This model can be easily loaded using the `AutoModelForCausalLM` functionality:
```python
>>> from transformers import pipeline
>>> generator = pipeline('text-generation', model='EleutherAI/gpt-j-6B')
>>> generator("EleutherAI has", do_sample=True, min_length=50)
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("gpt2")
model = AutoModelForCausalLM.from_pretrained("gpt-j-6B")
[{'generated_text': 'EleutherAI has made a commitment to create new software packages for each of its major clients and has'}]
```