An error about TransformerWeightGenerator

[Jack-bo1220]

encoder_layer = nn.TransformerEncoderLayer(
TypeError: init() got an unexpected keyword argument 'norm_first'

This will show an error after I delete it：

transformer_output = self.transformer_encoder(x)
tgt_len, bsz, embed_dim = query.size()
ValueError: not enough values to unpack (expected 3, got 2

I found that the shape of x in “transformer_output = self.transformer_encoder(x)” is torch.Size([132, 128])

[Jack-bo1220]

This seems to be highly related to the torch version, could you tell me the torch version used for the implementation?

[moonboy12138]

I tried 2.0.1+cu117 and it worked smoothly

[Jack-bo1220]

I got it. thank you so much