Skip to content

Commit e99a47f

Browse files
committed
common: add config presets for falcon
1 parent f5e96b3 commit e99a47f

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

common/arg.cpp

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3423,5 +3423,21 @@ common_params_context common_params_parser_init(common_params & params, llama_ex
34233423
}
34243424
).set_examples({LLAMA_EXAMPLE_SERVER}));
34253425

3426+
add_opt(common_arg(
3427+
{"--fim-falcon-7b-spec"},
3428+
string_format("use quantized Falcon 7B model (note: can download weights from the internet)"),
3429+
[](common_params & params) {
3430+
params.model.hf_repo = "maddes8cht/tiiuae-falcon-7b-gguf";
3431+
params.model.hf_file = "tiiuae-falcon-7b-Q6_K.gguf";
3432+
params.n_gpu_layers = 40;
3433+
params.flash_attn = true;
3434+
params.n_ubatch = 1024;
3435+
params.n_batch = 1024;
3436+
params.n_ctx = 2048;
3437+
params.n_cache_reuse = 256;
3438+
}
3439+
).set_examples({LLAMA_EXAMPLE_SERVER}));
3440+
3441+
34263442
return ctx_arg;
34273443
}

0 commit comments

Comments
 (0)