Follow up fix custom flash block sizes fallback in Ulysses attention#437

ninatu wants to merge 2 commits into

mainAI-Hypercomputer/maxdiffusion:mainfrom

ninatu/fix_heads_per_tile2AI-Hypercomputer/maxdiffusion:ninatu/fix_heads_per_tile2Copy head branch name to clipboard