Skip to content

Commit 0de614f

Browse files
committed
Fix tblock
1 parent 9d354fd commit 0de614f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/gpt2/layer/tblock.m

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ function outputHasInputSizeWithPasts(test,Input)
3737
% Provide a fake past of sequence length 1
3838
K_fake = dlarray(rand(C,1));
3939
V_fake = dlarray(rand(C,1));
40-
past = cat(4,K_fake,V_fake);
40+
past = cat(5,K_fake,V_fake);
4141
[y,present] = test.block(x,past,weights,hyperParameters);
4242
test.verifySize(y,size(x));
4343
% The size of presents is the size of past except the sequence

0 commit comments

Comments
 (0)