.gitignore
.pre-commit-config.yaml
LICENSE.md
MANIFEST.in
README.md
pyproject.toml
pytest.ini
quick_install.sh
requirements.txt
setup.cfg
setup.py
.github/workflows/python-publish.yml
.github/workflows/python_packages.yml
assets/flux_fbc_0.06.png
assets/flux_fbc_0.08.png
assets/flux_fbc_0.10.png
assets/flux_fbc_0.12.png
assets/flux_original.png
assets/hunyuan_video_fbc.mp4
assets/hunyuan_video_original.mp4
doc/fastest_flux.md
doc/fastest_hunyuan_video.md
first_block_cache_examples/run_cogvideox.py
first_block_cache_examples/run_flux.py
first_block_cache_examples/run_hunyuan_video.py
first_block_cache_examples/run_mochi.py
first_block_cache_examples/run_wan.py
parallel_examples/run_cogvideox.py
parallel_examples/run_flux.py
parallel_examples/run_hunyuan_video.py
parallel_examples/run_mochi.py
parallel_examples/run_wan.py
src/para_attn/__init__.py
src/para_attn/_version.py
src/para_attn/config.py
src/para_attn/para_attn_interface.py
src/para_attn/primitives.py
src/para_attn/sparse_attn.py
src/para_attn/utils.py
src/para_attn.egg-info/PKG-INFO
src/para_attn.egg-info/SOURCES.txt
src/para_attn.egg-info/dependency_links.txt
src/para_attn.egg-info/requires.txt
src/para_attn.egg-info/top_level.txt
src/para_attn/context_parallel/__init__.py
src/para_attn/context_parallel/diffusers_adapters/__init__.py
src/para_attn/context_parallel/diffusers_adapters/cogvideox.py
src/para_attn/context_parallel/diffusers_adapters/flux.py
src/para_attn/context_parallel/diffusers_adapters/hunyuan_video.py
src/para_attn/context_parallel/diffusers_adapters/mochi.py
src/para_attn/context_parallel/diffusers_adapters/wan.py
src/para_attn/distributed/__init__.py
src/para_attn/distributed/mp_runner.py
src/para_attn/first_block_cache/__init__.py
src/para_attn/first_block_cache/utils.py
src/para_attn/first_block_cache/diffusers_adapters/__init__.py
src/para_attn/first_block_cache/diffusers_adapters/cogvideox.py
src/para_attn/first_block_cache/diffusers_adapters/flux.py
src/para_attn/first_block_cache/diffusers_adapters/hunyuan_video.py
src/para_attn/first_block_cache/diffusers_adapters/mochi.py
src/para_attn/first_block_cache/diffusers_adapters/wan.py
src/para_attn/ops/__init__.py
src/para_attn/parallel_vae/__init__.py
src/para_attn/parallel_vae/diffusers_adapters/__init__.py
src/para_attn/parallel_vae/diffusers_adapters/autoencoder_kl.py
src/para_attn/parallel_vae/diffusers_adapters/autoencoder_kl_hunyuan_video.py
src/para_attn/parallel_vae/diffusers_adapters/autoencoder_kl_wan.py
tests/__init__.py
tests/conftest.py
tests/test_interface.py
tests/context_parallel/__init__.py
tests/context_parallel/test_diffusers_adapters.py
tests/ops/__init__.py
tests/ops/test_ops.py