Skip to content

Commit b28ad7e

Browse files
committed
fix attn weight scaling after rebase
1 parent 223edef commit b28ad7e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/llava/clip.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -778,7 +778,7 @@ static ggml_cgraph * clip_image_build_graph_legacy(clip_ctx * ctx, const clip_im
778778
if (full_attn) {
779779
KQ = ggml_soft_max_ext(ctx0, KQ, nullptr, 1.0f / sqrtf((float)d_head), 0.0f);
780780
} else {
781-
KQ = ggml_soft_max_ext(ctx0, KQ, window_mask, 1.0f, 0.0f);
781+
KQ = ggml_soft_max_ext(ctx0, KQ, window_mask, 1.0f / sqrtf((float)d_head), 0.0f);
782782
}
783783

784784
struct ggml_tensor * KQV = ggml_mul_mat(ctx0, V, KQ);

0 commit comments

Comments
 (0)