AssertionError: input tensor should be 1 dimensional.

[Fatima-Naseem071]

I am encountering the following error when I use a subset of VCTK dataset for training a multi-speaker TTS model using tacotron 2:

[!] Waveform is not finite everywhere. Skipping the GL.
Audio tensor shape before squeeze: (1,)
Audio tensor shape after squeeze: (1,)
! Run is kept in /home/cle-dl-09/TTS-dev/recipes/vctk/tacotron2/run-August-16-2024_03+46PM-20c8635
Traceback (most recent call last):
File "/home/cle-dl-09/TTS_Coqui/lib/python3.9/site-packages/trainer/trainer.py", line 1833, in fit
self._fit()
File "/home/cle-dl-09/TTS_Coqui/lib/python3.9/site-packages/trainer/trainer.py", line 1787, in _fit
self.eval_epoch()
File "/home/cle-dl-09/TTS_Coqui/lib/python3.9/site-packages/trainer/trainer.py", line 1665, in eval_epoch
self.model.eval_log(
File "/home/cle-dl-09/TTS_Coqui/lib/python3.9/site-packages/TTS/tts/models/tacotron2.py", line 417, in eval_log
logger.eval_audios(steps, audios, self.ap.sample_rate)
File "/home/cle-dl-09/TTS_Coqui/lib/python3.9/site-packages/trainer/logging/base_dash_logger.py", line 83, in eval_audios
self.add_audios(scope_name="EvalAudios", audios=audios, step=step, sample_rate=sample_rate)
File "/home/cle-dl-09/TTS_Coqui/lib/python3.9/site-packages/trainer/logging/tensorboard_logger.py", line 60, in add_audios
self.add_audio(
File "/home/cle-dl-09/TTS_Coqui/lib/python3.9/site-packages/trainer/logging/tensorboard_logger.py", line 36, in add_audio
self.writer.add_audio(title, audio, step, sample_rate=sample_rate)
File "/home/cle-dl-09/TTS_Coqui/lib/python3.9/site-packages/torch/utils/tensorboard/writer.py", line 798, in add_audio
audio(tag, snd_tensor, sample_rate=sample_rate), global_step, walltime
File "/home/cle-dl-09/TTS_Coqui/lib/python3.9/site-packages/torch/utils/tensorboard/summary.py", line 701, in audio
assert array.ndim == 1, "input tensor should be 1 dimensional."
AssertionError: input tensor should be 1 dimensional.

I am encountering this assertion error after varying number of epochs each time when I start training but my training never goes beyond 10 epochs.
I request someone to help me debug this issue.

Thank you!