use CHECK_ARG for arguemnt parsing

ggerganov · Jun 24, 2024 · 632a718 · 632a718
1 parent ff6d3c1
commit 632a718
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/common/common.cpp b/common/common.cpp
@@ -473,10 +473,7 @@ bool gpt_params_find_arg(int argc, char ** argv, const std::string & arg, gpt_pa
         return true;
     }
     if (arg == "--attention") {
-        if (++i >= argc) {
-            invalid_param = true;
-            return true;
-        }
+        CHECK_ARG
         std::string value(argv[i]);
         /**/ if (value == "causal") { params.attention_type = LLAMA_ATTENTION_TYPE_CAUSAL; }
         else if (value == "non-causal") { params.attention_type = LLAMA_ATTENTION_TYPE_NON_CAUSAL; }