Fix custom flash block sizes fallback in Ulysses attention#420

copybara-service[bot] merged 1 commit into

mainAI-Hypercomputer/maxdiffusion:mainfrom

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