UNQ_C6 assertion error: Wrong values in attn_w_b2. Check the call to self.mha2

I have checked self.mha2 call and I cant understand where I am doing my mistake, my code for out1 and block 1 and 2

mentor edit: code removed

the error message is:

Please do not post your code on the Forums. That breaks the course Honor Code.

You’re not supposed to add a dropout layer. The 2nd comment line under BLOCK 1 says this.