transformers
Fix torchscript tests for GPT-NeoX
#18012
Merged

Loading