fix

Picovoice · Apr 30, 2024 · a9b30c3 · a9b30c3
1 parent a04684e
commit a9b30c3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)