Fixes qwen2 too

This commit is contained in:
jukofyork 2024-06-22 11:07:17 +01:00 committed by GitHub
parent 8e092616a9
commit af2e033051
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -9442,6 +9442,13 @@ struct llm_build_context {
cb(cur, "ffn_out", il);
cur = ggml_add(ctx0, cur, ffn_inp);
// 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