search

project-baize / baize-chatbot

Let ChatGPT teach your own chatbot in hours with a single GPU!
https://arxiv.org/abs/2304.01196
GNU General Public License v3.0
3.16k stars 282 forks source link

Is this a typo in preprocess.py? #57

Open abdullin opened 1 year ago

abdullin commented 1 year ago

This line builds up the end of the prompt, which should be completed by LLM.

https://github.com/project-baize/baize-chatbot/blob/4fae6c4e550f087958c1f60746f0e6290e6b9f02/preprocess.py#L42C46-L42C46

            if s != "":
                prompt = "The conversation between human and AI assistant.\n"
                s = prompt + s + "[|Human|] "

Shouldn't the line end with [|AI|] and not [|HUMAN|]?