transformers
9df8b301 - Reenable SDPA's FA2 During Training with torch.compile (#30442)

Commit
1 year ago
Reenable SDPA's FA2 During Training with torch.compile (#30442) * Reenable SDPA's FA2 during training with torch.compile * fix Olmo's SDPA FA2 dispatching too * update formatting * improved SDPA comment * formatting and explanatory comment * is_causal if statement to one-liner
Parents
Loading