transformers
347a0f9e - Simplify GQA conditions in sdpa_attention.py (#41699)

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.
Author
Parents
Loading