Spaces:

AEUPH
/

GPT4ALL_CHAT

Runtime error

AEUPH commited on Aug 18, 2023

Commit

6da337d

1 Parent(s): 8f391b5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 class RPG_Game:
     def __init__(self):
-        self.model_name = "gpt4all/GPT4ALL"
         self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
         self.model = AutoModelForCausalLM.from_pretrained(self.model_name)
         self.inventory = {}

 class RPG_Game:
     def __init__(self):
+        self.model_name = "nomic-ai/gpt4all-j"
         self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
         self.model = AutoModelForCausalLM.from_pretrained(self.model_name)
         self.inventory = {}