The `build_command_r` forgot to add the control vector.
This commit is contained in:
jukofyork 2024-06-22 10:39:26 +01:00 committed by GitHub
parent c5a8d4b749
commit 8e092616a9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -11047,6 +11047,13 @@ struct llm_build_context {
// add together residual + FFN + self-attention
cur = ggml_add(ctx0, cur, inpL);
cur = ggml_add(ctx0, cur, attn_out);
// add control vector
ggml_tensor * layer_dir = lctx.cvec.tensor_for(il);
if (layer_dir != nullptr) {
cur = ggml_add(ctx0, cur, layer_dir);
}
cb(cur, "l_out", il);
// input for next layer