Skip to content

Commit

Permalink
llama: rwkv6: Apply code format changes
Browse files Browse the repository at this point in the history
Signed-off-by: Molly Sophia <[email protected]>
  • Loading branch information
MollySophia committed Aug 26, 2024
1 parent 59e8461 commit a1429c2
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/llama.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -15123,7 +15123,7 @@ struct llm_build_context {
)
);

ggml_tensor * x_norm_ffn = llm_build_norm(ctx0, cur, hparams, layer->attn_norm_2, layer->attn_norm_2_b, LLM_NORM, cb, il);
struct ggml_tensor * x_norm_ffn = llm_build_norm(ctx0, cur, hparams, layer->attn_norm_2, layer->attn_norm_2_b, LLM_NORM, cb, il);
x_prev = ggml_concat(
ctx0,
ffn_shift,
Expand Down Expand Up @@ -15159,7 +15159,7 @@ struct llm_build_context {
}

cur = inpL;
ggml_tensor * inp_out_ids = build_inp_out_ids();
struct ggml_tensor * inp_out_ids = build_inp_out_ids();
cur = ggml_reshape_2d(ctx0, cur, n_embd, n_tokens);
cur = ggml_get_rows(ctx0, cur, inp_out_ids);

Expand Down

0 comments on commit a1429c2

Please sign in to comment.