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

update chat message template

Author: mudler

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}}