[`FA-2`] Final fix for FA2 dtype (#26846)
* final fix for FA2 dtype
* try
* oops
* Update src/transformers/models/falcon/modeling_falcon.py
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
* apply fix everywhere
---------
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>