Skip to content
Unit Test

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

Sign in to view logs

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

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

Run time

Learn about OS pricing on GitHub Actions
Job Run time
TPU test (v4-8)
35m 45s
35m 45s