transformers
2cf87e2b
- Prevent Dynamo graph fragmentation in GPTNeoX with torch.baddbmm fix (#24941)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
Prevent Dynamo graph fragmentation in GPTNeoX with torch.baddbmm fix (#24941) * Pass a Python scalar for alpha in torch.baddbmm * fixup --------- Co-authored-by: Arthur Zucker <arthur.zucker@gmail.com>
References
#24941 - Prevent Dynamo graph fragmentation in GPTNeoX with torch.baddbmm fix
Author
norabelrose
Parents
b413e061
Loading