Update utils.py

utils.py

@@ -58,12 +58,12 @@ def generate_prompt_with_history(text, history, tokenizer, max_length=2048):
 
 
 
+#tokenizer = AutoTokenizer.from_pretrained("project-baize/baize-v2-7b")
+#model = AutoModelForCausalLM.from_pretrained("project-baize/baize-v2-7b")
+tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
+model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
 
 
-tokenizer = AutoTokenizer.from_pretrained("project-baize/baize-v2-7b")
-
-model = AutoModelForCausalLM.from_pretrained("project-baize/baize-v2-7b")
-
 def load_tokenizer_and_model(base_model,load_8bit=False):
     if torch.cuda.is_available():
         device = "cuda"