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