From 0f240bb2f931921d0a552cb16bb6845e45a244fc Mon Sep 17 00:00:00 2001 From: Herman Semenov Date: Mon, 26 Aug 2024 01:08:05 +0300 Subject: [PATCH] common: fixed not working find argument --n-gpu-layers-draft --- common/common.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/common/common.cpp b/common/common.cpp index 59e829660..d46df39b5 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()) {