RuntimeError: The expanded size of the tensor (30) must match the existing size (50) at non-singleton dimension 1. Target sizes: [50, 30, 50]. Tensor sizes: [50, 50]

[sunnyangazy]

i test the example code in colab ,but it has a error

[Gothos]

I've identified the issue, it's been fixed on the github repo, I'll update the pip package soon.