diff --git a/web_demo.py b/web_demo.py index 4091a98..26de0ba 100644 --- a/web_demo.py +++ b/web_demo.py @@ -46,7 +46,7 @@ def prepare_generation_config(): return generation_config -user_prompt = "<|User|>:{user}\n" +user_prompt = "<|User|>:{user}\n" robot_prompt = "<|Bot|>:{robot}\n" cur_query_prompt = "<|User|>:{user}\n<|Bot|>:"