common: add `LLAMA_ARG_OVERRIDE_TENSOR` env var for `-ot` arg (#18267)

diff --git a/common/arg.cpp b/common/arg.cpp
index 1302065498227c4e1c2d6958cc37354a86fa3b49..fded0bd260d59d3162f9e00074a787837be3e565 100644 (file)
--- a/common/arg.cpp
+++ b/common/arg.cpp
@@ -2087,7 +2087,7 @@ common_params_context common_params_parser_init(common_params & params, llama_ex
         "override tensor buffer type", [](common_params & params, const std::string & value) {
             parse_tensor_buffer_overrides(value, params.tensor_buft_overrides);
         }
-    ));
+    ).set_env("LLAMA_ARG_OVERRIDE_TENSOR"));
     add_opt(common_arg(
         {"-otd", "--override-tensor-draft"}, "<tensor name pattern>=<buffer type>,...",
         "override tensor buffer type for draft model", [](common_params & params, const std::string & value) {