Spaces:

openaccess-ai-collective
/

ggml-ui

Build error

winglian commited on May 15, 2023

Commit

dfe6be1

1 Parent(s): 24b11c7

fix chat prompts for vicuna

Files changed (1) hide show

tabbed.py CHANGED Viewed

@@ -32,7 +32,7 @@ def chat(history, system_message, max_tokens, temperature, top_p, top_k, repeat_
     history = history or []
     messages = system_message + \
-               "\n".join(["\n".join(["### User: "+item[0], "Assistant: "+item[1]])
                         for item in history])
     history[-1][1] = ""

     history = history or []
     messages = system_message + \
+               "\n".join(["\n".join(["USER: "+item[0], "ASSISTANT: "+item[1]])
                         for item in history])
     history[-1][1] = ""