Add tokenizer longhaul cache
This commit is contained in:
@@ -160,6 +160,21 @@ static void test(void) {
|
||||
assert(params.load_mode == LLAMA_LOAD_MODE_LONGHAUL);
|
||||
assert(params.longhaul_cache_bytes == 2ULL * 1024 * 1024 * 1024);
|
||||
|
||||
params.tokenizer_longhaul = false;
|
||||
params.tokenizer_longhaul_cache_bytes = 0;
|
||||
argv = {"binary_name", "--tokenizer-longhaul"};
|
||||
assert(false == common_params_parse(argv.size(), list_str_to_char(argv).data(), params, LLAMA_EXAMPLE_COMMON));
|
||||
|
||||
argv = {"binary_name", "--tokenizer-longhaul", "--tokenizer-longhaul-cache", "64"};
|
||||
assert(true == common_params_parse(argv.size(), list_str_to_char(argv).data(), params, LLAMA_EXAMPLE_COMMON));
|
||||
assert(params.tokenizer_longhaul);
|
||||
assert(params.tokenizer_longhaul_cache_bytes == 64ULL * 1024 * 1024);
|
||||
|
||||
params.tokenizer_longhaul = false;
|
||||
params.tokenizer_longhaul_cache_bytes = 0;
|
||||
argv = {"binary_name", "--tokenizer-longhaul-cache", "64"};
|
||||
assert(false == common_params_parse(argv.size(), list_str_to_char(argv).data(), params, LLAMA_EXAMPLE_COMMON));
|
||||
|
||||
argv = {"binary_name", "--token-cache-size", "256", "--token-cache-dir", "/tmp/llama-token-cache-test.sqlite3"};
|
||||
assert(true == common_params_parse(argv.size(), list_str_to_char(argv).data(), params, LLAMA_EXAMPLE_COMMON));
|
||||
assert(params.token_cache_size_mib == 256);
|
||||
|
||||
Reference in New Issue
Block a user