Просмотр исходного кода

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

Herman Semenov 1 год назад
Родитель
Сommit
93bc3839f9
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      common/common.cpp

+ 1 - 1
common/common.cpp

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