3 Commits

Author	SHA1	Message	Date
Dan Saunders	544f2a8a27	moving tests around for flash_attn install	2024-12-20 20:39:53 +00:00
Dan Saunders	d4e29e5b67	adding split_heads argument for retaining original (Q, K) dimensionanlity	2024-12-20 20:39:53 +00:00
Dan Saunders	80ba0d8dd1	isolating problematic test	2024-12-20 20:39:53 +00:00