[LTX-2] Fix flash attention shard_map for sequence lengths not divisible by context mesh axis#363

copybara-service[bot] merged 1 commit intomainAI-Hypercomputer/maxdiffusion:mainfrom

ltx2_pipeline_fixAI-Hypercomputer/maxdiffusion:ltx2_pipeline_fixCopy head branch name to clipboard