Switch to mrsteyk/openchatgpt-neox-125m

app.py

@@ -11,8 +11,8 @@ Assistant is a distilled language model trained by the community.<|STK_SP|>
 
 [User]"""
 
-tokenizer = AutoTokenizer.from_pretrained("mrsteyk/openchatgpt-neo-125m", use_fast=True)
-model = AutoModelForCausalLM.from_pretrained("mrsteyk/openchatgpt-neo-125m")
+tokenizer = AutoTokenizer.from_pretrained("mrsteyk/openchatgpt-neox-125m", use_fast=True)
+model = AutoModelForCausalLM.from_pretrained("mrsteyk/openchatgpt-neox-125m")
 
 def chat(inpt, max_new_tokens, top_k, top_p, temperature, repetition_penalty):
     inputs = tokenizer(f"{prologue}\n{inpt}<|STK_SP|>\n\n[Assistant]\n", return_tensors="pt").input_ids