Upcast Cross Attention Layer To Float32