runtime error

Exit code: 1. Reason: raining, mask=self.mask if not self.training else None) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/root/.pyenv/versions/3.12.13/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1778, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/root/.pyenv/versions/3.12.13/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1789, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/root/.pyenv/versions/3.12.13/lib/python3.12/site-packages/whisperspeech/modules.py", line 227, in forward x = x + self.cross_attn(lnx, x_positions, xa, xa_positions) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/root/.pyenv/versions/3.12.13/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1778, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/root/.pyenv/versions/3.12.13/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1789, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/root/.pyenv/versions/3.12.13/lib/python3.12/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] Traceback (most recent call last): File "/app/app.py", line 139, in <module> demo.launch(server_port=3000 if DEVEL else None) File "/root/.pyenv/versions/3.12.13/lib/python3.12/site-packages/gradio/blocks.py", line 2717, in launch raise Exception( Exception: Couldn't start the app because 'http://localhost:7860/gradio_api/startup-events' failed (code 500). Check your network or proxy settings to ensure localhost is accessible. █

Container logs: