Trying to run transformer attention model using Keras but keep getting TypeError: unsupported operand type(s) for +: 'NoneType' and 'int'

Question

I have created the following module for an attention model I am trying to run that will have two inputs so I can have a crossmodality attention.

import random