Test / index.html
Chirp's picture
Update index.html
f116d6b
raw
history blame
213 Bytes
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("KoboldAI/GPT-NeoX-20B-Erebus")
model = AutoModelForCausalLM.from_pretrained("KoboldAI/GPT-NeoX-20B-Erebus")