diff --git a/src/run.c b/src/run.c index dc3d35a..ad6d1e9 100644 --- a/src/run.c +++ b/src/run.c @@ -322,6 +322,9 @@ static const char* chatframe(const char* style, bool has_system) { } else if (strcmp(style, "qwen2") == 0) { return has_system ? "<|im_start|>system\n%s<|im_end|>\n<|im_start|>user\n%s<|im_end|>\n<|im_start|>assistant\n" : "\n<|im_start|>user\n%s<|im_end|>\n<|im_start|>assistant\n"; + } else if (strcmp(style, "gemma") == 0) { + return has_system ? "user\nSYSTEM: %s\n%s\nmodel\n" + : "user\n%s\nmodel\n"; } else { return has_system ? "[INST] <>\n%s\n<>\n\n%s [/INST]" : "[INST] %s [/INST]"; }