text mask issue while training decoder

[Sevixdd]

Hello, I get this error when I pass the text embedings to train the decoder and I'm quite stuck

Traceback (most recent call last):
File "/decoder_test.py", line 103, in
loss = trainer.forward(img, **forward_params, unet_number=unet_number, _device=device)
File "lib/python3.8/site-packages/dalle2_pytorch/trainer.py", line 107, in inner
out = fn(model, *args, **kwargs)
File "lib/python3.8/site-packages/dalle2_pytorch/trainer.py", line 723, in forward
loss_obj = self.decoder(*chunked_args, unet_number = unet_number, return_lowres_cond_image=return_lowres_cond_image, **chunked_kwargs)
File "lib/python3.8/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "lib/python3.8/site-packages/dalle2_pytorch/dalle2_pytorch.py", line 3268, in forward
losses = self.p_losses(unet, image, times, image_embed = image_embed, text_encodings = text_encodings, lowres_cond_img = lowres_cond_img, predict_x_start = predict_x_start, predict_v = predict_v, learned_variance = learned_variance, is_latent_diffusion = is_latent_diffusion, noise_scheduler = noise_scheduler, lowres_noise_level = lowres_noise_level)
File "lib/python3.8/site-packages/dalle2_pytorch/dalle2_pytorch.py", line 3049, in p_losses
unet_output = unet(
File "/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "lib/python3.8/site-packages/dalle2_pytorch/dalle2_pytorch.py", line 2255, in forward
text_mask = text_mask[:, :self.max_text_len]
IndexError: too many indices for tensor of dimension 1

Process finished with exit code 1