Skip to content

Commit 40dd868

Browse files
committed
Update
1 parent d48164a commit 40dd868

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dlib/dnn/transformer.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ namespace dlib
5151
multm_prev3<softmaxm<tril_mask<
5252
scale_weights<d_model / num_heads,
5353
multm_prev4<
54-
rope<query<seq_len, d_model, num_heads, skip2<
54+
rope<query<seq_len, d_model, num_heads, skip1<
5555
tag4<transpose<
5656
rope<key<seq_len, d_model, num_heads, skip2<
5757
tag3<value<seq_len, d_model, num_heads,
@@ -114,7 +114,7 @@ namespace dlib
114114
multm_prev3<softmaxm<tril_mask<
115115
scale_weights<d_model / num_heads,
116116
multm_prev4<
117-
query<num_heads, d_model, skip2<
117+
query<num_heads, d_model, skip1<
118118
tag4<key<num_heads, d_model, skip2<
119119
tag3<value<num_heads, d_model,
120120
tag2<fc_no_bias<d_model * 3,

0 commit comments

Comments
 (0)