]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commitdiff
common: fixed not working find argument --n-gpu-layers-draft (#9175)
authorHerman Semenov <redacted>
Sun, 25 Aug 2024 22:54:37 +0000 (22:54 +0000)
committerGitHub <redacted>
Sun, 25 Aug 2024 22:54:37 +0000 (00:54 +0200)
common/common.cpp

index 59e8296604c9c8cda9b781f8a7097cbc7ad69c2e..d46df39b5ac3cf98858f6331a831556eecec173e 100644 (file)
@@ -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()) {