diff options
| author | Minsoo Cheong <54794500+mscheong01@users.noreply.github.com> | 2024-03-25 16:38:22 +0900 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-03-25 09:38:22 +0200 |
| commit | 64e7b47c6986221f2ff5c57c89dfc018bb0e9e6d (patch) | |
| tree | 136cffc7261408388aca6a07fdd293263a19f34d /common/common.h | |
| parent | 7733f0c76081b2a69b5f8b192db2db7c43629d58 (diff) | |
examples : add "retrieval" (#6193)
* add `retrieval` example
* add README
* minor fixes
* cast filepos on print
* remove use of variable sized array
* store similarities in separate vector
* print error on insufficient batch size
* fix error message printing
* assign n_batch value to n_ubatch
* fix param definitions
* define retrieval-only parameters in retrieval.cpp
* fix `--context-file` option to be provided multiple times for multiple files
* use vector for `query_emb`
* add usage description in README
* fix merge conflict
* fix usage printing
* remove seed setting
* fix lint
* increase file read buffer size
* retrieval : minor
---------
Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
Diffstat (limited to 'common/common.h')
| -rw-r--r-- | common/common.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/common/common.h b/common/common.h index a223ecea..99ee90bc 100644 --- a/common/common.h +++ b/common/common.h @@ -171,6 +171,8 @@ bool gpt_params_parse(int argc, char ** argv, gpt_params & params); void gpt_print_usage(int argc, char ** argv, const gpt_params & params); +bool gpt_params_find_arg(int argc, char ** argv, const std::string & arg, gpt_params & params, int & i, bool & invalid_param); + std::string get_system_info(const gpt_params & params); std::string gpt_random_prompt(std::mt19937 & rng); |