File size: 213 Bytes
f116d6b
 
 
 
 
1
2
3
4
5
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("KoboldAI/GPT-NeoX-20B-Erebus")

model = AutoModelForCausalLM.from_pretrained("KoboldAI/GPT-NeoX-20B-Erebus")