1 files changed, 4 insertions, 0 deletions
diff --git a/common/common.cpp b/common/common.cpp
index 0dbde58f..2df8d4d4 100644
--- a/common/common.cpp
+++ b/common/common.cpp
@@ -1468,6 +1468,10 @@ bool gpt_params_find_arg(int argc, char ** argv, const std::string & arg, gpt_pa
         params.warmup = false;
         return true;
     }
+    if (arg == "--warmup-batch" || arg == "-wb") {
+        params.batch_warmup = true;
+        return true;
+    }
     if (arg == "--output-format") {
         CHECK_ARG
         std::string value(argv[i]);