runtime error

Exit code: 1. Reason: le.py", line 1747, in _call_impl return forward_call(*args, **kwargs) File "/home/user/.pyenv/versions/3.10.15/lib/python3.10/site-packages/whisperspeech/t2s_up_wds_mlang_enclm.py", line 337, in forward x = self.decoder(x, in_stoks_positions, xenc.clone(), xenc_positions) File "/home/user/.pyenv/versions/3.10.15/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "/home/user/.pyenv/versions/3.10.15/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl return forward_call(*args, **kwargs) File "/home/user/.pyenv/versions/3.10.15/lib/python3.10/site-packages/whisperspeech/modules.py", line 250, in forward x = l(x, x_positions, xenc, xenc_positions, causal=self.training, mask=self.mask if not self.training else None) File "/home/user/.pyenv/versions/3.10.15/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "/home/user/.pyenv/versions/3.10.15/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl return forward_call(*args, **kwargs) File "/home/user/.pyenv/versions/3.10.15/lib/python3.10/site-packages/whisperspeech/modules.py", line 227, in forward x = x + self.cross_attn(lnx, x_positions, xa, xa_positions) File "/home/user/.pyenv/versions/3.10.15/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "/home/user/.pyenv/versions/3.10.15/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl return forward_call(*args, **kwargs) File "/home/user/.pyenv/versions/3.10.15/lib/python3.10/site-packages/whisperspeech/modules.py", line 139, in forward self.k_cache[:k.shape[0],:,kv_positions] = k RuntimeError: shape mismatch: value tensor of shape [550, 12, 550, 64] cannot be broadcast to indexing result of shape [1, 12, 550, 64] █

Container logs: