Fix torchscript tests for GPT-NeoX#18012

LysandreJik merged 4 commits intohuggingface:mainhuggingface/transformers:mainfrom

ydshieh:fix_torchscript_tests_for_gpt_neoxydshieh/transformers:fix_torchscript_tests_for_gpt_neoxCopy head branch name to clipboard