quantize: override fix build

phymbert · Apr 20, 2024 · 66a4b6a · 66a4b6a
1 parent deba48c
commit 66a4b6a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/common/common.cpp b/common/common.cpp
@@ -266,7 +266,7 @@ bool parse_kv_override(const char * data, std::vector<llama_model_kv_override> &
     } else if (strncmp(sep, "str:", 4) == 0) {
         sep += 4;
         kvo.tag = LLAMA_KV_OVERRIDE_TYPE_STR;
-        strncpy(kvo.str_value, sep, 128);
+        strncpy(kvo.str_value, sep, 127);
     } else {
         fprintf(stderr, "%s: invalid type for KV override '%s'\n", __func__, data);
         return false;

diff --git a/examples/quantize/quantize.cpp b/examples/quantize/quantize.cpp
@@ -305,14 +305,14 @@ int main(int argc, char ** argv) {
             llama_model_kv_override kvo;
             std::strcpy(kvo.key, LLM_KV_QUANTIZE_IMATRIX_FILE);
             kvo.tag = LLAMA_KV_OVERRIDE_TYPE_STR;
-            strncpy(kvo.str_value, imatrix_file.c_str(), 128);
+            strncpy(kvo.str_value, imatrix_file.c_str(), 127);
             kv_overrides.emplace_back(std::move(kvo));
         }
         if (!imatrix_dataset.empty()) {
             llama_model_kv_override kvo;
             std::strcpy(kvo.key, LLM_KV_QUANTIZE_IMATRIX_DATASET);
             kvo.tag = LLAMA_KV_OVERRIDE_TYPE_STR;
-            strncpy(kvo.str_value, imatrix_dataset.c_str(), 128);
+            strncpy(kvo.str_value, imatrix_dataset.c_str(), 127);
             kv_overrides.emplace_back(std::move(kvo));
         }