transformers
347a0f9e
- Simplify GQA conditions in sdpa_attention.py (#41699)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
189 days ago
Simplify GQA conditions in sdpa_attention.py (#41699) Removed unnecessary checks for key being a torch.fx.Proxy in GQA conditions because fx tracing is no longer supported, and torch.export supports enable_gqa.
References
#41699 - Simplify GQA conditions in sdpa_attention.py
Author
justinchuby
Parents
7e204ad1
Loading