diff --git a/examples/mario/Modelfile b/examples/mario/Modelfile index 9bb271bc..02759e87 100644 --- a/examples/mario/Modelfile +++ b/examples/mario/Modelfile @@ -1,7 +1,11 @@ FROM llama2 PARAMETER temperature 1 PROMPT """ -System: You are Mario from super mario bros, acting as an assistant. -User: {{ .Prompt }} -Assistant: -""" +{{- if not .Context }} +<> +You are Mario from super mario bros, acting as an assistant. +<> + +{{- end }} +[INST] {{ .Prompt }} [/INST] +""" \ No newline at end of file diff --git a/examples/mario/logo.png b/examples/mario/logo.png index 935dde9e..1ef25646 100644 Binary files a/examples/mario/logo.png and b/examples/mario/logo.png differ diff --git a/examples/mario/readme.md b/examples/mario/readme.md index 23c77efc..1e1a6889 100644 --- a/examples/mario/readme.md +++ b/examples/mario/readme.md @@ -21,9 +21,13 @@ What the model file looks like: FROM llama2 PARAMETER temperature 1 PROMPT """ -System: You are Mario from super mario bros, acting as an assistant. -User: {{ .Prompt }} -Assistant: +{{- if not .Context }} +<> +You are Mario from super mario bros, acting as an assistant. +<> + +{{- end }} +[INST] {{ .Prompt }} [/INST] """ ```