fix extra colon in InternLMChat7B (#796)

bd7c4e39 · Qian Zhao · GitHub · 079f29bc · bd7c4e39
Unverified Commit bd7c4e39 authored Dec 05, 2023 by Qian Zhao Committed by GitHub Dec 05, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lmdeploy/model.py lmdeploy/model.py +1 -1

No files found.
--- a/lmdeploy/model.py
+++ b/lmdeploy/model.py
@@ -238,7 +238,7 @@ class InternLMChat7B(BaseModel):
            role = message['role']
            content = message['content']
            ret += f'{eval(f"self.{role}")}{content}{eox_map[role]}'
-        ret += f'{self.assistant}:'
+        ret += f'{self.assistant}'
        return ret