Skip to content

Commit 55a485e

Browse files
GermanAizekNexesenex
authored andcommitted
common: fixed not working find argument --n-gpu-layers-draft (ggml-org#9175)
1 parent 4d7cabd commit 55a485e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

common/common.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -906,7 +906,7 @@ bool gpt_params_find_arg(int argc, char ** argv, const std::string & arg, gpt_pa
906906
}
907907
return true;
908908
}
909-
if (arg == "-ngld" || arg == "--gpu-layers-draft" || arg == "--gpu-layers-draft") {
909+
if (arg == "-ngld" || arg == "--gpu-layers-draft" || arg == "--n-gpu-layers-draft") {
910910
CHECK_ARG
911911
params.n_gpu_layers_draft = std::stoi(argv[i]);
912912
if (!llama_supports_gpu_offload()) {

0 commit comments

Comments
 (0)