Skip to content

Commit a55386b

Browse files
committed
Maintain batch dimension in preprocessor.
Fixes bug when batch size > 1.
1 parent a7c0d33 commit a55386b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/layers/moonshine.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ namespace ctranslate2 {
2020

2121
StorageView input(output_type(), features.device());
2222
StorageView input_reshaped = std::move(features);
23-
input_reshaped.expand_dims(0);
23+
input_reshaped.expand_dims(1);
2424

2525
_conv1(input_reshaped, input);
2626
_tanh(input, input);

0 commit comments

Comments
 (0)