vrr/koboldcpp
2
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2025-09-11 09:34:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

server : enable -td and -tbd parameters (#15172)

Browse source
This commit is contained in:
Sigbjørn Skjæret 2025-08-13 15:43:00 +02:00 • committed by GitHub
parent c24f4e2688
commit b3e16665e1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
common/arg.cpp
View file

@ -3163,7 +3163,7 @@ common_params_context common_params_parser_init(common_params & params, llama_ex
params.speculative.cpuparams.n_threads = std::thread::hardware_concurrency(); params.speculative.cpuparams.n_threads = std::thread::hardware_concurrency();
} }
} }
).set_examples({LLAMA_EXAMPLE_SPECULATIVE})); ).set_examples({LLAMA_EXAMPLE_SPECULATIVE, LLAMA_EXAMPLE_SERVER}));
add_opt(common_arg( add_opt(common_arg(
{"-tbd", "--threads-batch-draft"}, "N", {"-tbd", "--threads-batch-draft"}, "N",
"number of threads to use during batch and prompt processing (default: same as --threads-draft)", "number of threads to use during batch and prompt processing (default: same as --threads-draft)",
@ -3173,7 +3173,7 @@ common_params_context common_params_parser_init(common_params & params, llama_ex
params.speculative.cpuparams_batch.n_threads = std::thread::hardware_concurrency(); params.speculative.cpuparams_batch.n_threads = std::thread::hardware_concurrency();
} }
} }
).set_examples({LLAMA_EXAMPLE_SPECULATIVE})); ).set_examples({LLAMA_EXAMPLE_SPECULATIVE, LLAMA_EXAMPLE_SERVER}));
add_opt(common_arg( add_opt(common_arg(
{"-Cd", "--cpu-mask-draft"}, "M", {"-Cd", "--cpu-mask-draft"}, "M",
"Draft model CPU affinity mask. Complements cpu-range-draft (default: same as --cpu-mask)", "Draft model CPU affinity mask. Complements cpu-range-draft (default: same as --cpu-mask)",

2
tools/server/server.cpp
View file

@ -2015,6 +2015,8 @@ struct server_context {
params_dft.cache_type_k = params_base.speculative.cache_type_k; params_dft.cache_type_k = params_base.speculative.cache_type_k;
params_dft.cache_type_v = params_base.speculative.cache_type_v; params_dft.cache_type_v = params_base.speculative.cache_type_v;
params_dft.cpuparams.n_threads = params_base.speculative.cpuparams.n_threads;
params_dft.cpuparams_batch.n_threads = params_base.speculative.cpuparams_batch.n_threads;
params_dft.tensor_buft_overrides = params_base.speculative.tensor_buft_overrides; params_dft.tensor_buft_overrides = params_base.speculative.tensor_buft_overrides;
llama_init_dft = common_init_from_params(params_dft); llama_init_dft = common_init_from_params(params_dft);