From 3639e4702feaff4f8e1a58643c6b2d1a3a261e55 Mon Sep 17 00:00:00 2001
From: Xu Song <xusong.vip@gmail.com>
Date: Thu, 4 May 2023 22:08:09 +0800
Subject: [PATCH] Fix prompt

---
 moss_cli_demo.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/moss_cli_demo.py b/moss_cli_demo.py
index bb36e40..5dfe175 100644
--- a/moss_cli_demo.py
+++ b/moss_cli_demo.py
@@ -90,7 +90,7 @@ def main():
                 eos_token_id=106068,
                 pad_token_id=tokenizer.pad_token_id)
             response = tokenizer.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_tokens=True)
-            prompt += response
+            prompt += response + '<eom>'
             print(response.lstrip('\n'))
     
 if __name__ == "__main__":