Selaa lähdekoodia

updating prompt

Michael Chiang 1 vuosi sitten
vanhempi
commit
e54c08da89
3 muutettua tiedostoa jossa 15 lisäystä ja 7 poistoa
  1. 8 4
      examples/mario/Modelfile
  2. BIN
      examples/mario/logo.png
  3. 7 3
      examples/mario/readme.md

+ 8 - 4
examples/mario/Modelfile

@@ -1,7 +1,11 @@
 FROM llama2
 FROM llama2
 PARAMETER temperature 1
 PARAMETER temperature 1
 PROMPT """
 PROMPT """
-System: You are Mario from super mario bros, acting as an assistant.
-User: {{ .Prompt }}
-Assistant:
-"""
+{{- if not .Context }}
+<<SYS>>
+You are Mario from super mario bros, acting as an assistant.
+<</SYS>>
+
+{{- end }}
+[INST] {{ .Prompt }} [/INST]
+"""

BIN
examples/mario/logo.png


+ 7 - 3
examples/mario/readme.md

@@ -21,9 +21,13 @@ What the model file looks like:
 FROM llama2
 FROM llama2
 PARAMETER temperature 1
 PARAMETER temperature 1
 PROMPT """
 PROMPT """
-System: You are Mario from super mario bros, acting as an assistant.
-User: {{ .Prompt }}
-Assistant:
+{{- if not .Context }}
+<<SYS>>
+You are Mario from super mario bros, acting as an assistant.
+<</SYS>>
+
+{{- end }}
+[INST] {{ .Prompt }} [/INST]
 """
 """
 ```
 ```