diff options
author | anzz1 <anzz1@live.com> | 2023-03-21 17:42:43 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-03-21 17:42:43 +0200 |
commit | 975d2cebf97ce888fa0aeee6f5ac774d7135891f (patch) | |
tree | e578d57ca7ccef7851e1f02dfe15887ff829aec4 /utils.cpp | |
parent | e0ffc861fae5ac8b40ce973f822d03db02929d36 (diff) |
cmdline option for custom amount of model parts (--n_parts N) (#348)
* cmdline option for custom amount of model parts (--n_parts N)
* Update main.cpp
---------
Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
Diffstat (limited to 'utils.cpp')
-rw-r--r-- | utils.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
@@ -74,6 +74,8 @@ bool gpt_params_parse(int argc, char ** argv, gpt_params & params) { params.antiprompt.push_back(argv[++i]); } else if (arg == "--ignore-eos") { params.ignore_eos = true; + } else if (arg == "--n_parts") { + params.n_parts = std::stoi(argv[++i]); } else if (arg == "-h" || arg == "--help") { gpt_print_usage(argc, argv, params); exit(0); @@ -116,6 +118,7 @@ void gpt_print_usage(int /*argc*/, char ** argv, const gpt_params & params) { fprintf(stderr, " --ignore-eos ignore end of stream token and continue generating\n"); fprintf(stderr, " --memory_f16 use f16 instead of f32 for memory key+value\n"); fprintf(stderr, " --temp N temperature (default: %.1f)\n", params.temp); + fprintf(stderr, " --n_parts N number of model parts (default: -1 = determine from dimensions)\n"); fprintf(stderr, " -b N, --batch_size N batch size for prompt processing (default: %d)\n", params.n_batch); fprintf(stderr, " -m FNAME, --model FNAME\n"); fprintf(stderr, " model path (default: %s)\n", params.model.c_str()); |