An extra dropout layer?

[VincentPisztora]

Hi there - thank you for posting this repository - it has been very helpful. In the model.py file, if I'm reading it right, there are two back-to-back dropout layers in the TransformerBlock call. The first is the last layer of the "mlp" layer (line 67) called on line 82 and the second is the "dropout2" layer (line 73) which is called on line 83. Should one of the two be removed? Thanks!