Skip to content

Commit a4774ce

Browse files
update modeling_mim.py
1 parent 4357cfb commit a4774ce

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

mindnlp/transformers/models/mimi/modeling_mimi.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -188,6 +188,7 @@ def _pad1d(hidden_states: ms.Tensor, paddings: Tuple[int, int], mode: str = "zer
188188
"""
189189
length = hidden_states.shape[-1]
190190
padding_left, padding_right = paddings
191+
paddings = (int(padding_left), int(padding_right))
191192
if not mode == "reflect":
192193
# "ConstantPadND()(input=<Tensor>, padding=<list of int, Tensor, tuple of int>, value=<Number>)".
193194
return nn.functional.pad(hidden_states, paddings, mode, value)

0 commit comments

Comments
 (0)