Skip to content

Commit 93bc383

Browse files
authored
common: fixed not working find argument --n-gpu-layers-draft (ggml-org#9175)
1 parent f91fc56 commit 93bc383

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
@@ -901,7 +901,7 @@ bool gpt_params_find_arg(int argc, char ** argv, const std::string & arg, gpt_pa
901901
}
902902
return true;
903903
}
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") {
905905
CHECK_ARG
906906
params.n_gpu_layers_draft = std::stoi(argv[i]);
907907
if (!llama_supports_gpu_offload()) {

0 commit comments

Comments
 (0)