server: validate "--chat-template" argument

ngxson · Feb 9, 2024 · ebe3079 · ebe3079
1 parent 7efef47
commit ebe3079
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/examples/server/server.cpp b/examples/server/server.cpp
@@ -2300,7 +2300,13 @@ static void server_params_parse(int argc, char **argv, server_params &sparams,
                 invalid_param = true;
                 break;
             }
-            sparams.chat_template = argv[i];
+            std::string value(argv[i]);
+            if (value != "chatml" && value != "llama2") {
+                fprintf(stderr, "error: chat template can be \"llama2\" or \"chatml\", but got: %s\n", value.c_str());
+                invalid_param = true;
+                break;
+            }
+            sparams.chat_template = value;
         }
         else if (arg == "--override-kv")
         {