llama : use n_embd_gqa instead of n_embd to handle llama-2 70B (#2433)
This commit is contained in:
parent
edcc7ae7d2
commit
65cdf34bdc
2 changed files with 3 additions and 2 deletions
|
@ -26,6 +26,7 @@ int main(int argc, char ** argv) {
|
|||
auto lparams = llama_context_default_params();
|
||||
|
||||
lparams.n_ctx = params.n_ctx;
|
||||
lparams.n_gqa = params.n_gqa;
|
||||
lparams.seed = params.seed;
|
||||
lparams.f16_kv = params.memory_f16;
|
||||
lparams.use_mmap = params.use_mmap;
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue