runtime error

cks.py", line 1012, in process_api result = await self.call_function( File "/home/user/.local/lib/python3.10/site-packages/gradio/blocks.py", line 830, in call_function prediction = await anyio.to_thread.run_sync( File "/home/user/.local/lib/python3.10/site-packages/anyio/to_thread.py", line 33, in run_sync return await get_asynclib().run_sync_in_worker_thread( File "/home/user/.local/lib/python3.10/site-packages/anyio/_backends/_asyncio.py", line 877, in run_sync_in_worker_thread return await future File "/home/user/.local/lib/python3.10/site-packages/anyio/_backends/_asyncio.py", line 807, in run result = context.run(func, *args) File "/home/user/app/app.py", line 54, in predict speech = model.generate_speech(input_ids, speaker_embedding, vocoder=vocoder) File "/home/user/.local/lib/python3.10/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context return func(*args, **kwargs) File "/home/user/.local/lib/python3.10/site-packages/transformers/models/speecht5/modeling_speecht5.py", line 2921, in generate_speech return _generate_speech( File "/home/user/.local/lib/python3.10/site-packages/transformers/models/speecht5/modeling_speecht5.py", line 2521, in _generate_speech decoder_hidden_states = model.speecht5.decoder.prenet(output_sequence, speaker_embeddings) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl return forward_call(*args, **kwargs) File "/home/user/.local/lib/python3.10/site-packages/transformers/models/speecht5/modeling_speecht5.py", line 700, in forward speaker_embeddings = speaker_embeddings.expand(-1, inputs_embeds.size(1), -1) RuntimeError: expand(torch.FloatTensor{[1, 1, 1, 512]}, size=[-1, 1, -1]): the number of sizes provided (3) must be greater or equal to the number of dimensions in the tensor (4)

Container logs: