from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("KoboldAI/GPT-NeoX-20B-Erebus") model = AutoModelForCausalLM.from_pretrained("KoboldAI/GPT-NeoX-20B-Erebus")