Unverified Commit baf1801b authored by AllentDan's avatar AllentDan Committed by GitHub
Browse files

update solar chat template (#587)

parent 186bfd2e
...@@ -585,9 +585,9 @@ class SOLAR(BaseModel): ...@@ -585,9 +585,9 @@ class SOLAR(BaseModel):
def __init__(self, def __init__(self,
b_sys='### System:\n', b_sys='### System:\n',
e_sys='\n\n', e_sys='\n\n',
boh='### User:\n', user='### User:\n',
eoh='\n\n', eoh='\n\n',
boa='### Assistant:\n', assistant='### Assistant:\n',
eoa='\n\n', eoa='\n\n',
system='', system='',
session_len=2048, session_len=2048,
...@@ -595,9 +595,9 @@ class SOLAR(BaseModel): ...@@ -595,9 +595,9 @@ class SOLAR(BaseModel):
super().__init__(**kwargs) super().__init__(**kwargs)
self.b_sys = b_sys self.b_sys = b_sys
self.e_sys = e_sys self.e_sys = e_sys
self.boh = boh self.user = user
self.eoh = eoh self.eoh = eoh
self.boa = boa self.assistant = assistant
self.eoa = eoa self.eoa = eoa
self.system = system self.system = system
self.session_len = session_len self.session_len = session_len
...@@ -617,9 +617,9 @@ class SOLAR(BaseModel): ...@@ -617,9 +617,9 @@ class SOLAR(BaseModel):
f'{type(self).__name__} has no capability of {self.capability}' f'{type(self).__name__} has no capability of {self.capability}'
if sequence_start: if sequence_start:
return f'{self.b_sys}{self.system}{self.e_sys}' \ return f'{self.b_sys}{self.system}{self.e_sys}' \
f'{self.boh}{prompt}{self.eoh}{self.boa}' f'{self.user}{prompt}{self.eoh}{self.assistant}'
return f'{self.boh}{prompt} {self.eoh}{self.boa}' return f'{self.user}{prompt}{self.eoh}{self.assistant}'
def messages2prompt(self, messages, sequence_start=True): def messages2prompt(self, messages, sequence_start=True):
"""Return the prompt that is concatenated with other elements in the """Return the prompt that is concatenated with other elements in the
...@@ -636,7 +636,7 @@ class SOLAR(BaseModel): ...@@ -636,7 +636,7 @@ class SOLAR(BaseModel):
system = self.system if not system else system system = self.system if not system else system
ret = f'{self.b_sys}{system}{self.e_sys}' ret = f'{self.b_sys}{system}{self.e_sys}'
for i, (user, assistant) in enumerate(zip(users, assistants)): for i, (user, assistant) in enumerate(zip(users, assistants)):
ret += f'{self.boh}{user}{self.eoh}{self.boa}' ret += f'{self.user}{user}{self.eoh}{self.assistant}'
if assistant: if assistant:
ret += f'{assistant}{self.eoa}' ret += f'{assistant}{self.eoa}'
return ret return ret
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment