Skip to content

Commit 4edfd20

Browse files
author
Alejandro Gaston Alvarez Franceschi
committed
More fixes
1 parent 0d3238a commit 4edfd20

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

coremltools/converters/mil/mil/passes/defs/lower_complex_dialect_ops.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -511,11 +511,11 @@ def _overlap_add(
511511

512512
# Create empty output with final shape
513513
if input_shape.rank == 3:
514-
channels, n_frames = input_shape.val
515-
output = mb.fill(shape=(channels, int(n_fft.val + hop_length.val * (n_frames - 1))), value=0., before_op=before_op)
514+
channels, n_frames, _= input_shape.val
515+
output = mb.fill(shape=(channels, int(n_fft.val + hop_length.val * (n_frames - 1)),), value=0., before_op=before_op)
516516
else:
517517
channels = None
518-
n_frames= input_shape.val
518+
n_frames, _ = input_shape.val
519519
output = mb.fill(shape=(int(n_fft.val + hop_length.val * (n_frames - 1)),), value=0., before_op=before_op)
520520

521521
# Create an index used later on overlap add

0 commit comments

Comments
 (0)