bugfix
This commit is contained in:
parent
652349f4fa
commit
ff802a1755
@ -34,7 +34,7 @@ class BaseReranker:
|
|||||||
def build_sys_prompt(self, sys_prompt):
|
def build_sys_prompt(self, sys_prompt):
|
||||||
return f"<|im_start|>system\n{sys_prompt}\n<|im_end|>"
|
return f"<|im_start|>system\n{sys_prompt}\n<|im_end|>"
|
||||||
|
|
||||||
def build_user_prompt(self, query, docs, instruct=''):
|
def build_user_prompt(self, query, doc, instruct=''):
|
||||||
return f'<|im_start|>user\n<Instruct>: {instruct}\n<Query>:{query}\n<Document>:\n{doc}<|im_end|>'
|
return f'<|im_start|>user\n<Instruct>: {instruct}\n<Query>:{query}\n<Document>:\n{doc}<|im_end|>'
|
||||||
|
|
||||||
def build_assistant_prompt(self):
|
def build_assistant_prompt(self):
|
||||||
|
Loading…
Reference in New Issue
Block a user