Fix custom flash block sizes fallback in Ulysses attention#420

ninatu wants to merge 1 commit into

mainAI-Hypercomputer/maxdiffusion:mainfrom

ninatu/fix_heads_per_tileAI-Hypercomputer/maxdiffusion:ninatu/fix_heads_per_tileCopy head branch name to clipboard