Compare commits

...
Sign in to create a new pull request.

1 commit

Author SHA1 Message Date
Francis Couture-Harpin
64b7d85891 llama : fix command-r inference 2024-03-28 06:22:24 -04:00

View file

@ -9154,6 +9154,7 @@ struct llm_build_context {
struct ggml_tensor * inp_out_ids = build_inp_out_ids();
cur = ggml_get_rows(ctx0, cur, inp_out_ids);
inpL = ggml_get_rows(ctx0, inpL, inp_out_ids);
ffn_inp = ggml_get_rows(ctx0, ffn_inp, inp_out_ids);
}
struct ggml_tensor * attn_out = cur;