W4 programming Assigment

Despite passing all the test cases for Transformer architecture with tensorflow assignment (w4 programming assignment) I am seeing issues where grader is returning 25%.

The grader should give you feedback about which parts of the assignment do not work correctly. Please post a screen capture image showing the grader feedback.