transformers

ac98a88f - Fix torchscript tests for GPT-NeoX (#18012)

Fix torchscript tests for GPT-NeoX (#18012) * fix dtype issue in _attn * fix RotaryEmbedding * fix RotaryEmbedding 2 * clean up Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>