yuxiaohui78

> Same issue here~ Replace the torch.view_as_complex with the code below in your model. ``` def view_as_complex(real_tensor): real_part = real_tensor[..., 0] imag_part = real_tensor[..., 1] return torch.complex(real_part, imag_part) ```