diff --git a/binding/python/_picollm.py b/binding/python/_picollm.py index 52676552..f9310fbe 100644 --- a/binding/python/_picollm.py +++ b/binding/python/_picollm.py @@ -121,7 +121,7 @@ def prompt(self) -> str: res.append(f"<|start_header_id|>user<|end_header_id|>\n\n{h.strip()}<|eot_id|>") res.append(f"<|start_header_id|>assistant<|end_header_id|>\n\n{l.strip()}<|eot_id|>") res.append(f"<|start_header_id|>user<|end_header_id|>\n\n{human[-1].strip()}<|eot_id|>") - res.append(f"<|start_header_id|>assistant<|end_header_id|>\n\n") + res.append("<|start_header_id|>assistant<|end_header_id|>\n\n") return ''.join(res) @@ -298,7 +298,7 @@ def __str__(self) -> str: for x in self.top_choices: tc = f""" {{ token: {repr(x.token)}, - log-prob: {x.log_prob:.2f}, + log-prob: {x.log_prob:.2f}, }}""" tcs.append(tc) tcs = ',\n'.join(tcs)