diff --git a/gpt4all-training/README.md b/gpt4all-training/README.md index f57955de..d595478c 100644 --- a/gpt4all-training/README.md +++ b/gpt4all-training/README.md @@ -23,7 +23,7 @@ from datasets import load_dataset from transformers import AutoModelForCausalLM dataset = load_dataset("nomic-ai/gpt4all-j-prompt-generations", revision="v1.2-jazzy") -model = AutoModelForCausalLM.from_pretrained("nomic-ai/gpt4all-j-prompt-generations", revision="v1.2-jazzy") +model = AutoModelForCausalLM.from_pretrained("nomic-ai/gpt4all-j", revision="v1.2-jazzy") ``` ### GPT4All-J Training Instructions