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

copybara-service[bot] merged 1 commit into

mainAI-Hypercomputer/maxdiffusion:mainfrom

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