Blog-writer / README.md
imseldrith's picture
Create README.md
be423ca
|
Raw
History Blame Contribute Delete
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")