Blog-writer / README.md
imseldrith's picture
Create README.md
be423ca
|
raw
history blame
203 Bytes

from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B")

model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-neo-1.3B")