Issue with running Tactron2 with Pytorch2

Hi all,
trying to use torch compile with Tacotron model, but I always get that error (‘lengths’ argument should be a 1D CPU int64 tensor, but got 1D meta Long tensor), that fired in forward function when calling x = nn.utils.rnn.pack_padded_sequence()
even when converting input_lengths to CPU.