greg / gptj.py
gregtsen's picture
Create gptj.py
1f36a5e
raw
history blame contribute delete
193 Bytes
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("nomic-ai/gpt4all-j")
model = AutoModelForCausalLM.from_pretrained("nomic-ai/gpt4all-j")