PR #18012 Fix torchscript tests for GPT-NeoX

Fix torchscript tests for GPT-NeoX #18012

LysandreJik merged 4 commits into huggingface:main from ydshieh:fix_torchscript_tests_for_gpt_neox

fix dtype issue in _attn

5c9adf36

fix RotaryEmbedding

32394f85

fix RotaryEmbedding 2

89b37624

ydshieh requested a review from

sgugger 3 years ago

ydshieh requested a review from

patil-suraj 3 years ago

ydshieh requested a review from

patrickvonplaten 3 years ago

sgugger approved these changes on 2022-07-05

clean up

d9befca9

patrickvonplaten approved these changes on 2022-07-07

LysandreJik approved these changes on 2022-07-11

LysandreJik merged ac98a88f into main 3 years ago

ydshieh deleted the fix_torchscript_tests_for_gpt_neox branch 3 years ago

Reviewers

LysandreJik

patrickvonplaten

sgugger

patil-suraj

Assignees

No one assigned

Labels

None yet

Milestone

No milestone