update chat message template

go-skynet · Jul 31, 2023 · 5890977 · 5890977
1 parent 0ffecda
commit 5890977
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/llama2-chat.yaml b/llama2-chat.yaml
@@ -26,8 +26,7 @@ prompt_templates:
   content: |
     {{if eq .RoleName "assistant"}}{{.Content}}{{else}}
     [INST]
-    {{if .SystemPrompt}}{{.SystemPrompt}}{{else if eq .RoleName "system"}}<<SYS>>{{.Content}}<</SYS>>
-
-    {{else if .Content}}{{.Content}}{{end}}
+    {{if eq .RoleName "system"}}<<SYS>>{{.Content}}<</SYS>>{{else if and (.SystemPrompt) (eq .MessageIndex 0)}}<<SYS>>{{.SystemPrompt}}<</SYS>>{{end}}
+    {{if .Content}}{{.Content}}{{end}}
     [/INST] 
     {{end}}