Skip to content

Commit

Permalink
Maintain batch dimension in preprocessor.
Browse files Browse the repository at this point in the history
Fixes bug when batch size > 1.
  • Loading branch information
njeffrie committed Nov 26, 2024
1 parent a7c0d33 commit a55386b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/layers/moonshine.cc
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ namespace ctranslate2 {

StorageView input(output_type(), features.device());
StorageView input_reshaped = std::move(features);
input_reshaped.expand_dims(0);
input_reshaped.expand_dims(1);

_conv1(input_reshaped, input);
_tanh(input, input);
Expand Down

0 comments on commit a55386b

Please sign in to comment.