PaddleNLP
PaddleNLP copied to clipboard
[GPTJ] add gptj
PR types
New features
PR changes
Models
Description
- Add new model GPTJ
- Use example
import paddle
from paddlenlp.transformers import GPTJForCausalLM, GPTJTokenizer
tokenizer = GPTJTokenizer.from_pretrained('EleutherAI/gpt-j-6B')
model = GPTJForCausalLM.from_pretrained('EleutherAI/gpt-j-6B')
inputs = tokenizer("Welcome to use PaddlePaddle and PaddleNLP!", return_token_type_ids=False)
inputs = {k:paddle.to_tensor([v]) for (k, v) in inputs.items()}
outputs = model.generate(**inputs)
Mv to #3017