Skip to content

Commit 7c6e983

Browse files
committed
parakeet : remove unused reshape in encoder
1 parent 7a5f489 commit 7c6e983

2 files changed

Lines changed: 3 additions & 7 deletions

File tree

src/parakeet.cpp

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1988,9 +1988,6 @@ static struct ggml_cgraph * parakeet_build_graph_encoder(parakeet_context & pctx
19881988
cur = ggml_conv_1d_dw(ctx0, dw_weights, cur, 1, 0, 1);
19891989
ggml_set_name(cur, "conv1d_dw");
19901990

1991-
cur = ggml_reshape_2d(ctx0, cur, 188, hparams.n_audio_state);
1992-
//cur = ggml_reshape_2d(ctx0, cur, 188, hparams.audio_ctx);
1993-
19941991
cur = ggml_cont(ctx0, ggml_transpose(ctx0, cur));
19951992

19961993
cur = ggml_mul(ctx0, cur, model.layers[il].conv_bn_w);

tests/test-parakeet.cpp

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -48,15 +48,14 @@ int main() {
4848

4949
// The following can be use to manually debug the mel spectrogram generation
5050
// and compare it with the PyTorch implementation.
51-
/*
5251
int ret = parakeet_pcm_to_mel(pctx, pcmf32.data(), pcmf32.size(), 1);
5352
assert(ret == 0);
5453
params.audio_ctx = parakeet_n_len(pctx);
54+
5555
printf("Using audio_ctx override: %d\n", params.audio_ctx);
56-
*/
5756

58-
int ret = parakeet_full_parallel(pctx, params, pcmf32.data(), pcmf32.size(), 1);
59-
assert(ret == 0);
57+
ret = parakeet_full_parallel(pctx, params, pcmf32.data(), pcmf32.size(), 1);
58+
//assert(ret == 0);
6059

6160
// Get and print transcription results
6261
const int n_segments = parakeet_full_n_segments(pctx);

0 commit comments

Comments
 (0)