MPT : fix an offload func typo

This commit is contained in:
Cebtenzzre 2023-10-14 13:00:39 -04:00
parent 2a4bcbacea
commit b577e6374e

View file

@ -5532,7 +5532,7 @@ static struct ggml_cgraph * llm_build_mpt(
wsize * n_embd_head, wsize * n_embd_head,
wsize * n_embd_head * (n_head + 2 * n_head_kv), wsize * n_embd_head * (n_head + 2 * n_head_kv),
wsize * n_embd_head * (n_head + n_head_kv)); wsize * n_embd_head * (n_head + n_head_kv));
offload_func_kq(Kcur); offload_func_v(tmpv);
ggml_set_name(Qcur, "Qcur"); ggml_set_name(Qcur, "Qcur");
ggml_set_name(Kcur, "Kcur"); ggml_set_name(Kcur, "Kcur");