From adfd9150d76cf1bfd4c7e4889fa7e79bcf21f9ca Mon Sep 17 00:00:00 2001
From: csunny <cfqsunny@163.com>
Date: Thu, 1 Jun 2023 15:26:36 +0800
Subject: [PATCH] debug: add info

---
 pilot/model/guanaco_llm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pilot/model/guanaco_llm.py b/pilot/model/guanaco_llm.py
index b765c87ca..03f2d1687 100644
--- a/pilot/model/guanaco_llm.py
+++ b/pilot/model/guanaco_llm.py
@@ -8,7 +8,7 @@ def guanaco_generate_output(model, tokenizer, params, device, context_len=2048):
 
     print(params)
     stop = params.get("stop", "###")
-    messages = params["prompt"].split(stop)
+    messages = params["prompt"]
 
 
     hist = []