Merge pull request #18 from dave-gray101/llama2-chat-template

update chat message template
go-skynet · Jul 31, 2023 · 09b798c · 09b798c
2 parents 0ffecda + 5890977
commit 09b798c
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}}