From: Herman Semenov <redacted>
Date: Sun, 25 Aug 2024 22:54:37 +0000 (+0000)
Subject: common: fixed not working find argument --n-gpu-layers-draft (#9175)
X-Git-Tag: upstream/0.0.4488~865
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=93bc3839f980ff14be86efe408b4cd7e89b26835;p=pkg%2Fggml%2Fsources%2Fllama.cpp

common: fixed not working find argument --n-gpu-layers-draft (#9175)
---

diff --git a/common/common.cpp b/common/common.cpp
index 59e82966..d46df39b 100644
--- a/common/common.cpp
+++ b/common/common.cpp
@@ -901,7 +901,7 @@ bool gpt_params_find_arg(int argc, char ** argv, const std::string & arg, gpt_pa
         }
         return true;
     }
-    if (arg == "-ngld" || arg == "--gpu-layers-draft" || arg == "--gpu-layers-draft") {
+    if (arg == "-ngld" || arg == "--gpu-layers-draft" || arg == "--n-gpu-layers-draft") {
         CHECK_ARG
         params.n_gpu_layers_draft = std::stoi(argv[i]);
         if (!llama_supports_gpu_offload()) {