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

GermanAizek · arthw · commit 4b8b910eef7b · 2024-11-18T13:51:05.000+08:00
diff --git 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()) {

Original file line number	Diff line number	Diff line change
`@@ -901,7 +901,7 @@ bool gpt_params_find_arg(int argc, char ** argv, const std::string & arg, gpt_pa`
`901`	`901`	`}`
`902`	`902`	`return true;`
`903`	`903`	`}`
`904`		`- if (arg == "-ngld" \|\| arg == "--gpu-layers-draft" \|\| arg == "--gpu-layers-draft") {`
	`904`	`+ if (arg == "-ngld" \|\| arg == "--gpu-layers-draft" \|\| arg == "--n-gpu-layers-draft") {`
`905`	`905`	`CHECK_ARG`
`906`	`906`	`params.n_gpu_layers_draft = std::stoi(argv[i]);`
`907`	`907`	`if (!llama_supports_gpu_offload()) {`