[`GPT-J`] Fix causal mask dtype (#23147)

Commit

2 years ago

[`GPT-J`] Fix causal mask dtype (#23147) * fix #23136 * better fix * same fix for `masked_bias`