transformers
Prevent Dynamo graph fragmentation in GPTNeoX with torch.baddbmm fix
#24941

Merged

Prevent Dynamo graph fragmentation in GPTNeoX with torch.baddbmm fix #24941

ArthurZucker merged 3 commits into huggingface:main from norabelrose:neox-norm-fix

Pass a Python scalar for alpha in torch.baddbmm

6339cf69

ArthurZucker commented on 2023-07-20

Merge branch 'main' into neox-norm-fix

38b3cbda

fixup

414cb5b7

ArthurZucker approved these changes on 2023-08-23

ArthurZucker requested a review from

sgugger 2 years ago

sgugger approved these changes on 2023-08-23

ArthurZucker merged 2cf87e2b into main 2 years ago

Reviewers

ArthurZucker

sgugger

Assignees

No one assigned

Labels

None yet

Milestone

No milestone