├── .github └── workflows │ └── python-publish.yml ├── .gitignore ├── .gitmodules ├── .pre-commit-config.yaml ├── LICENSE.txt ├── README.md ├── assets ├── DiTLevelCache-interp.png ├── FastCache-Video.png ├── architecture.png ├── logo.jpeg ├── overview.png ├── sample_caching.png ├── sample_org.png └── stat_thresh_plot.pdf ├── benchmark ├── README_cache.md ├── cache_execute.py ├── cache_execution_xfuser.py ├── fastcache_benchmark.py ├── fid │ ├── README.md │ ├── compute_fid.py │ ├── flux_generate.py │ ├── generate.sh │ ├── pixartalpha_generate.py │ └── process_ref_data.py ├── run.sh ├── single_node_latency_test.py ├── usp_latency_test.py └── video_cache_execute.py ├── cache_benchmark_results ├── benchmark_results.json ├── fast_image.png ├── fb_image.png ├── none_image.png └── tea_image.png ├── cache_execute_results ├── cache_execute_results.json ├── fast_image.png ├── fb_image.png ├── none_image.png └── tea_image.png ├── cache_results ├── cache_execute_results.json ├── fast_image.png ├── fb_image.png ├── none_image.png └── tea_image.png ├── docker └── Dockerfile ├── docs ├── developer │ ├── Http_Service.md │ ├── The_implement_design_of_xdit_framework.md │ ├── The_implement_design_of_xdit_framework_zh.md │ └── adding_models │ │ ├── adding_model_cfg.md │ │ ├── adding_model_cfg.py │ │ ├── adding_model_cfg_usp.md │ │ ├── adding_model_cfg_usp.py │ │ ├── adding_model_pipefusion.md │ │ ├── adding_model_pipefusion_zh.md │ │ ├── adding_model_usp.md │ │ ├── adding_model_usp.py │ │ ├── adding_model_usp_text_replica.md │ │ ├── adding_model_usp_text_replica.py │ │ └── readme.md ├── fid │ └── FID.md ├── methods │ ├── cfg_parallel.md │ ├── cfg_parallel_zh.md │ ├── ditfastattn.md │ ├── ditfastattn_zh.md │ ├── fastcache.md │ ├── hybrid.md │ ├── hybrid_zh.md │ ├── parallel_vae.md │ ├── pipefusion.md │ ├── quickstart.md │ └── usp.md └── performance │ ├── cogvideo.md │ ├── cogvideo_zh.md │ ├── consisid.md │ ├── consisid_zh.md │ ├── flux.md │ ├── flux_zh.md │ ├── hunyuandit.md │ ├── hunyuandit_zh.md │ ├── hunyuanvideo.md │ ├── latte.md │ ├── latte_zh.md │ ├── pixart_alpha_legacy.md │ ├── sd3.md │ ├── sd3_zh.md │ ├── stepvideo.md │ └── stepvideo_zh.md ├── entrypoints ├── __init__.py ├── curl.sh └── launch.py ├── examples ├── cogvideox_example.py ├── cogvideox_usp_example.py ├── consisid_example.py ├── consisid_usp_example.py ├── fastcache_benchmark.py ├── fastcache_benchmark_simple.py ├── fastcache_simple_test.py ├── fixed_cogvideox_distributed.py ├── fixed_cogvideox_example.py ├── flux_example.py ├── flux_usp_example.py ├── generate_benchmark_report.py ├── hunyuan_video_usp_example.py ├── hunyuandit_example.py ├── latte_example.py ├── minimal_cogvideo.py ├── pixartalpha_example.py ├── pixartsigma_example.py ├── ray │ ├── README.md │ ├── ray_flux_example.py │ ├── ray_hunyuandit_example.py │ ├── ray_pixartalpha_example.py │ ├── ray_pixartsigma_example.py │ ├── ray_run.sh │ └── ray_sd3_example.py ├── run.sh ├── run_cache_comparison.sh ├── run_cogvideo.sh ├── run_consisid.sh ├── run_consisid_usp.sh ├── run_fastcache_benchmark.sh ├── run_fastcache_simple.sh ├── run_fastcache_test.py ├── run_fastcache_test.sh ├── run_fastditattn.sh ├── run_hunyuan_video_usp.sh ├── run_multinodes.sh ├── run_pixart_fastcache.sh ├── run_service.sh ├── sd3_example.py ├── sdxl_example.py ├── simple_fastcache_video.py ├── simple_video_gen.py ├── test_enhanced_linear_approx.py └── test_fastcache_standalone.py ├── pytest.ini ├── setup.py ├── tests ├── context_parallel │ ├── debug_flux_usp_example.py │ ├── debug_tests.py │ └── test_diffusers_adapters.py ├── core │ ├── test_ring_flash_attn.py │ └── test_xfuser_attn.py ├── layers │ ├── attention_processor_test.py │ └── feedforward_test.py └── parallel_test.py ├── video_cache_results └── video_cache_results.json └── xfuser ├── __init__.py ├── __version__.py ├── config ├── __init__.py ├── args.py └── config.py ├── core ├── __init__.py ├── cache_manager │ ├── __init__.py │ └── cache_manager.py ├── distributed │ ├── __init__.py │ ├── group_coordinator.py │ ├── parallel_state.py │ ├── runtime_state.py │ └── utils.py ├── fast_attention │ ├── __init__.py │ ├── attn_layer.py │ ├── fast_attn_state.py │ └── utils.py ├── long_ctx_attention │ ├── __init__.py │ ├── hybrid │ │ ├── __init__.py │ │ └── attn_layer.py │ └── ring │ │ ├── __init__.py │ │ └── ring_flash_attn.py └── utils │ ├── __init__.py │ └── timer.py ├── envs.py ├── logger.py ├── model_executor ├── __init__.py ├── accelerator │ └── fastcache.py ├── base_wrapper.py ├── cache │ ├── __init__.py │ ├── diffusers_adapters │ │ ├── __init__.py │ │ ├── flux.py │ │ ├── pixart.py │ │ └── registry.py │ └── utils.py ├── layers │ ├── __init__.py │ ├── attention_processor.py │ ├── base_layer.py │ ├── conv.py │ ├── embeddings.py │ ├── feedforward.py │ ├── register.py │ ├── usp.py │ └── usp_legacy.py ├── models │ ├── __init__.py │ ├── base_model.py │ ├── customized │ │ ├── __init__.py │ │ └── step_video_t2v │ │ │ ├── __init__.py │ │ │ ├── attentions.py │ │ │ ├── blocks.py │ │ │ ├── linear.py │ │ │ ├── normalization.py │ │ │ ├── rope.py │ │ │ └── tp_applicator.py │ └── transformers │ │ ├── __init__.py │ │ ├── base_transformer.py │ │ ├── cogvideox_transformer_3d.py │ │ ├── consisid_transformer_3d.py │ │ ├── hunyuan_transformer_2d.py │ │ ├── latte_transformer_3d.py │ │ ├── pixart_transformer_2d.py │ │ ├── register.py │ │ ├── transformer_flux.py │ │ └── transformer_sd3.py ├── patch │ ├── __init__.py │ └── unet_patch.py ├── pipelines │ ├── __init__.py │ ├── base_pipeline.py │ ├── dit_pipeline.py │ ├── fastcache_pipeline.py │ ├── pipeline_cogvideox.py │ ├── pipeline_consisid.py │ ├── pipeline_flux.py │ ├── pipeline_hunyuandit.py │ ├── pipeline_latte.py │ ├── pipeline_pixart_alpha.py │ ├── pipeline_pixart_sigma.py │ ├── pipeline_stable_diffusion_3.py │ ├── pipeline_stable_diffusion_xl.py │ └── register.py └── schedulers │ ├── __init__.py │ ├── base_scheduler.py │ ├── register.py │ ├── scheduling_ddim.py │ ├── scheduling_ddim_cogvideox.py │ ├── scheduling_ddpm.py │ ├── scheduling_dpm_cogvideox.py │ ├── scheduling_dpmsolver_multistep.py │ └── scheduling_flow_match_euler_discrete.py ├── parallel.py └── ray ├── __init__.py ├── pipeline ├── __init__.py ├── base_executor.py ├── pipeline_utils.py └── ray_utils.py └── worker ├── __init__.py ├── utils.py ├── worker.py └── worker_wrappers.py /.github/workflows/python-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/.github/workflows/python-publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/README.md -------------------------------------------------------------------------------- /assets/DiTLevelCache-interp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/assets/DiTLevelCache-interp.png -------------------------------------------------------------------------------- /assets/FastCache-Video.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/assets/FastCache-Video.png -------------------------------------------------------------------------------- /assets/architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/assets/architecture.png -------------------------------------------------------------------------------- /assets/logo.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/assets/logo.jpeg -------------------------------------------------------------------------------- /assets/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/assets/overview.png -------------------------------------------------------------------------------- /assets/sample_caching.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/assets/sample_caching.png -------------------------------------------------------------------------------- /assets/sample_org.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/assets/sample_org.png -------------------------------------------------------------------------------- /assets/stat_thresh_plot.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/assets/stat_thresh_plot.pdf -------------------------------------------------------------------------------- /benchmark/README_cache.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/benchmark/README_cache.md -------------------------------------------------------------------------------- /benchmark/cache_execute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/benchmark/cache_execute.py -------------------------------------------------------------------------------- /benchmark/cache_execution_xfuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/benchmark/cache_execution_xfuser.py -------------------------------------------------------------------------------- /benchmark/fastcache_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/benchmark/fastcache_benchmark.py -------------------------------------------------------------------------------- /benchmark/fid/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/benchmark/fid/README.md -------------------------------------------------------------------------------- /benchmark/fid/compute_fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/benchmark/fid/compute_fid.py -------------------------------------------------------------------------------- /benchmark/fid/flux_generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/benchmark/fid/flux_generate.py -------------------------------------------------------------------------------- /benchmark/fid/generate.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/benchmark/fid/generate.sh -------------------------------------------------------------------------------- /benchmark/fid/pixartalpha_generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/benchmark/fid/pixartalpha_generate.py -------------------------------------------------------------------------------- /benchmark/fid/process_ref_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/benchmark/fid/process_ref_data.py -------------------------------------------------------------------------------- /benchmark/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/benchmark/run.sh -------------------------------------------------------------------------------- /benchmark/single_node_latency_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/benchmark/single_node_latency_test.py -------------------------------------------------------------------------------- /benchmark/usp_latency_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/benchmark/usp_latency_test.py -------------------------------------------------------------------------------- /benchmark/video_cache_execute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/benchmark/video_cache_execute.py -------------------------------------------------------------------------------- /cache_benchmark_results/benchmark_results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/cache_benchmark_results/benchmark_results.json -------------------------------------------------------------------------------- /cache_benchmark_results/fast_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/cache_benchmark_results/fast_image.png -------------------------------------------------------------------------------- /cache_benchmark_results/fb_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/cache_benchmark_results/fb_image.png -------------------------------------------------------------------------------- /cache_benchmark_results/none_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/cache_benchmark_results/none_image.png -------------------------------------------------------------------------------- /cache_benchmark_results/tea_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/cache_benchmark_results/tea_image.png -------------------------------------------------------------------------------- /cache_execute_results/cache_execute_results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/cache_execute_results/cache_execute_results.json -------------------------------------------------------------------------------- /cache_execute_results/fast_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/cache_execute_results/fast_image.png -------------------------------------------------------------------------------- /cache_execute_results/fb_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/cache_execute_results/fb_image.png -------------------------------------------------------------------------------- /cache_execute_results/none_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/cache_execute_results/none_image.png -------------------------------------------------------------------------------- /cache_execute_results/tea_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/cache_execute_results/tea_image.png -------------------------------------------------------------------------------- /cache_results/cache_execute_results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/cache_results/cache_execute_results.json -------------------------------------------------------------------------------- /cache_results/fast_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/cache_results/fast_image.png -------------------------------------------------------------------------------- /cache_results/fb_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/cache_results/fb_image.png -------------------------------------------------------------------------------- /cache_results/none_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/cache_results/none_image.png -------------------------------------------------------------------------------- /cache_results/tea_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/cache_results/tea_image.png -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docs/developer/Http_Service.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/developer/Http_Service.md -------------------------------------------------------------------------------- /docs/developer/The_implement_design_of_xdit_framework.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/developer/The_implement_design_of_xdit_framework.md -------------------------------------------------------------------------------- /docs/developer/The_implement_design_of_xdit_framework_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/developer/The_implement_design_of_xdit_framework_zh.md -------------------------------------------------------------------------------- /docs/developer/adding_models/adding_model_cfg.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/developer/adding_models/adding_model_cfg.md -------------------------------------------------------------------------------- /docs/developer/adding_models/adding_model_cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/developer/adding_models/adding_model_cfg.py -------------------------------------------------------------------------------- /docs/developer/adding_models/adding_model_cfg_usp.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/developer/adding_models/adding_model_cfg_usp.md -------------------------------------------------------------------------------- /docs/developer/adding_models/adding_model_cfg_usp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/developer/adding_models/adding_model_cfg_usp.py -------------------------------------------------------------------------------- /docs/developer/adding_models/adding_model_pipefusion.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/developer/adding_models/adding_model_pipefusion.md -------------------------------------------------------------------------------- /docs/developer/adding_models/adding_model_pipefusion_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/developer/adding_models/adding_model_pipefusion_zh.md -------------------------------------------------------------------------------- /docs/developer/adding_models/adding_model_usp.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/developer/adding_models/adding_model_usp.md -------------------------------------------------------------------------------- /docs/developer/adding_models/adding_model_usp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/developer/adding_models/adding_model_usp.py -------------------------------------------------------------------------------- /docs/developer/adding_models/adding_model_usp_text_replica.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/developer/adding_models/adding_model_usp_text_replica.md -------------------------------------------------------------------------------- /docs/developer/adding_models/adding_model_usp_text_replica.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/developer/adding_models/adding_model_usp_text_replica.py -------------------------------------------------------------------------------- /docs/developer/adding_models/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/developer/adding_models/readme.md -------------------------------------------------------------------------------- /docs/fid/FID.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/fid/FID.md -------------------------------------------------------------------------------- /docs/methods/cfg_parallel.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/methods/cfg_parallel.md -------------------------------------------------------------------------------- /docs/methods/cfg_parallel_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/methods/cfg_parallel_zh.md -------------------------------------------------------------------------------- /docs/methods/ditfastattn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/methods/ditfastattn.md -------------------------------------------------------------------------------- /docs/methods/ditfastattn_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/methods/ditfastattn_zh.md -------------------------------------------------------------------------------- /docs/methods/fastcache.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/methods/fastcache.md -------------------------------------------------------------------------------- /docs/methods/hybrid.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/methods/hybrid.md -------------------------------------------------------------------------------- /docs/methods/hybrid_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/methods/hybrid_zh.md -------------------------------------------------------------------------------- /docs/methods/parallel_vae.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/methods/parallel_vae.md -------------------------------------------------------------------------------- /docs/methods/pipefusion.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/methods/pipefusion.md -------------------------------------------------------------------------------- /docs/methods/quickstart.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/methods/quickstart.md -------------------------------------------------------------------------------- /docs/methods/usp.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/methods/usp.md -------------------------------------------------------------------------------- /docs/performance/cogvideo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/performance/cogvideo.md -------------------------------------------------------------------------------- /docs/performance/cogvideo_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/performance/cogvideo_zh.md -------------------------------------------------------------------------------- /docs/performance/consisid.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/performance/consisid.md -------------------------------------------------------------------------------- /docs/performance/consisid_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/performance/consisid_zh.md -------------------------------------------------------------------------------- /docs/performance/flux.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/performance/flux.md -------------------------------------------------------------------------------- /docs/performance/flux_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/performance/flux_zh.md -------------------------------------------------------------------------------- /docs/performance/hunyuandit.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/performance/hunyuandit.md -------------------------------------------------------------------------------- /docs/performance/hunyuandit_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/performance/hunyuandit_zh.md -------------------------------------------------------------------------------- /docs/performance/hunyuanvideo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/performance/hunyuanvideo.md -------------------------------------------------------------------------------- /docs/performance/latte.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/performance/latte.md -------------------------------------------------------------------------------- /docs/performance/latte_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/performance/latte_zh.md -------------------------------------------------------------------------------- /docs/performance/pixart_alpha_legacy.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/performance/pixart_alpha_legacy.md -------------------------------------------------------------------------------- /docs/performance/sd3.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/performance/sd3.md -------------------------------------------------------------------------------- /docs/performance/sd3_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/performance/sd3_zh.md -------------------------------------------------------------------------------- /docs/performance/stepvideo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/performance/stepvideo.md -------------------------------------------------------------------------------- /docs/performance/stepvideo_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/docs/performance/stepvideo_zh.md -------------------------------------------------------------------------------- /entrypoints/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /entrypoints/curl.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/entrypoints/curl.sh -------------------------------------------------------------------------------- /entrypoints/launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/entrypoints/launch.py -------------------------------------------------------------------------------- /examples/cogvideox_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/cogvideox_example.py -------------------------------------------------------------------------------- /examples/cogvideox_usp_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/cogvideox_usp_example.py -------------------------------------------------------------------------------- /examples/consisid_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/consisid_example.py -------------------------------------------------------------------------------- /examples/consisid_usp_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/consisid_usp_example.py -------------------------------------------------------------------------------- /examples/fastcache_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/fastcache_benchmark.py -------------------------------------------------------------------------------- /examples/fastcache_benchmark_simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/fastcache_benchmark_simple.py -------------------------------------------------------------------------------- /examples/fastcache_simple_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/fastcache_simple_test.py -------------------------------------------------------------------------------- /examples/fixed_cogvideox_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/fixed_cogvideox_distributed.py -------------------------------------------------------------------------------- /examples/fixed_cogvideox_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/fixed_cogvideox_example.py -------------------------------------------------------------------------------- /examples/flux_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/flux_example.py -------------------------------------------------------------------------------- /examples/flux_usp_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/flux_usp_example.py -------------------------------------------------------------------------------- /examples/generate_benchmark_report.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/generate_benchmark_report.py -------------------------------------------------------------------------------- /examples/hunyuan_video_usp_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/hunyuan_video_usp_example.py -------------------------------------------------------------------------------- /examples/hunyuandit_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/hunyuandit_example.py -------------------------------------------------------------------------------- /examples/latte_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/latte_example.py -------------------------------------------------------------------------------- /examples/minimal_cogvideo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/minimal_cogvideo.py -------------------------------------------------------------------------------- /examples/pixartalpha_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/pixartalpha_example.py -------------------------------------------------------------------------------- /examples/pixartsigma_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/pixartsigma_example.py -------------------------------------------------------------------------------- /examples/ray/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/ray/README.md -------------------------------------------------------------------------------- /examples/ray/ray_flux_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/ray/ray_flux_example.py -------------------------------------------------------------------------------- /examples/ray/ray_hunyuandit_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/ray/ray_hunyuandit_example.py -------------------------------------------------------------------------------- /examples/ray/ray_pixartalpha_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/ray/ray_pixartalpha_example.py -------------------------------------------------------------------------------- /examples/ray/ray_pixartsigma_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/ray/ray_pixartsigma_example.py -------------------------------------------------------------------------------- /examples/ray/ray_run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/ray/ray_run.sh -------------------------------------------------------------------------------- /examples/ray/ray_sd3_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/ray/ray_sd3_example.py -------------------------------------------------------------------------------- /examples/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/run.sh -------------------------------------------------------------------------------- /examples/run_cache_comparison.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/run_cache_comparison.sh -------------------------------------------------------------------------------- /examples/run_cogvideo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/run_cogvideo.sh -------------------------------------------------------------------------------- /examples/run_consisid.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/run_consisid.sh -------------------------------------------------------------------------------- /examples/run_consisid_usp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/run_consisid_usp.sh -------------------------------------------------------------------------------- /examples/run_fastcache_benchmark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/run_fastcache_benchmark.sh -------------------------------------------------------------------------------- /examples/run_fastcache_simple.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/run_fastcache_simple.sh -------------------------------------------------------------------------------- /examples/run_fastcache_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/run_fastcache_test.py -------------------------------------------------------------------------------- /examples/run_fastcache_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/run_fastcache_test.sh -------------------------------------------------------------------------------- /examples/run_fastditattn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/run_fastditattn.sh -------------------------------------------------------------------------------- /examples/run_hunyuan_video_usp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/run_hunyuan_video_usp.sh -------------------------------------------------------------------------------- /examples/run_multinodes.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/run_multinodes.sh -------------------------------------------------------------------------------- /examples/run_pixart_fastcache.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/run_pixart_fastcache.sh -------------------------------------------------------------------------------- /examples/run_service.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/run_service.sh -------------------------------------------------------------------------------- /examples/sd3_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/sd3_example.py -------------------------------------------------------------------------------- /examples/sdxl_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/sdxl_example.py -------------------------------------------------------------------------------- /examples/simple_fastcache_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/simple_fastcache_video.py -------------------------------------------------------------------------------- /examples/simple_video_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/simple_video_gen.py -------------------------------------------------------------------------------- /examples/test_enhanced_linear_approx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/test_enhanced_linear_approx.py -------------------------------------------------------------------------------- /examples/test_fastcache_standalone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/examples/test_fastcache_standalone.py -------------------------------------------------------------------------------- /pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/pytest.ini -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/setup.py -------------------------------------------------------------------------------- /tests/context_parallel/debug_flux_usp_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/tests/context_parallel/debug_flux_usp_example.py -------------------------------------------------------------------------------- /tests/context_parallel/debug_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/tests/context_parallel/debug_tests.py -------------------------------------------------------------------------------- /tests/context_parallel/test_diffusers_adapters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/tests/context_parallel/test_diffusers_adapters.py -------------------------------------------------------------------------------- /tests/core/test_ring_flash_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/tests/core/test_ring_flash_attn.py -------------------------------------------------------------------------------- /tests/core/test_xfuser_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/tests/core/test_xfuser_attn.py -------------------------------------------------------------------------------- /tests/layers/attention_processor_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/tests/layers/attention_processor_test.py -------------------------------------------------------------------------------- /tests/layers/feedforward_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/tests/layers/feedforward_test.py -------------------------------------------------------------------------------- /tests/parallel_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/tests/parallel_test.py -------------------------------------------------------------------------------- /video_cache_results/video_cache_results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/video_cache_results/video_cache_results.json -------------------------------------------------------------------------------- /xfuser/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/__init__.py -------------------------------------------------------------------------------- /xfuser/__version__.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.4.3.post3" 2 | -------------------------------------------------------------------------------- /xfuser/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/config/__init__.py -------------------------------------------------------------------------------- /xfuser/config/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/config/args.py -------------------------------------------------------------------------------- /xfuser/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/config/config.py -------------------------------------------------------------------------------- /xfuser/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/core/__init__.py -------------------------------------------------------------------------------- /xfuser/core/cache_manager/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/core/cache_manager/__init__.py -------------------------------------------------------------------------------- /xfuser/core/cache_manager/cache_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/core/cache_manager/cache_manager.py -------------------------------------------------------------------------------- /xfuser/core/distributed/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/core/distributed/__init__.py -------------------------------------------------------------------------------- /xfuser/core/distributed/group_coordinator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/core/distributed/group_coordinator.py -------------------------------------------------------------------------------- /xfuser/core/distributed/parallel_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/core/distributed/parallel_state.py -------------------------------------------------------------------------------- /xfuser/core/distributed/runtime_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/core/distributed/runtime_state.py -------------------------------------------------------------------------------- /xfuser/core/distributed/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/core/distributed/utils.py -------------------------------------------------------------------------------- /xfuser/core/fast_attention/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/core/fast_attention/__init__.py -------------------------------------------------------------------------------- /xfuser/core/fast_attention/attn_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/core/fast_attention/attn_layer.py -------------------------------------------------------------------------------- /xfuser/core/fast_attention/fast_attn_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/core/fast_attention/fast_attn_state.py -------------------------------------------------------------------------------- /xfuser/core/fast_attention/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/core/fast_attention/utils.py -------------------------------------------------------------------------------- /xfuser/core/long_ctx_attention/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/core/long_ctx_attention/__init__.py -------------------------------------------------------------------------------- /xfuser/core/long_ctx_attention/hybrid/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/core/long_ctx_attention/hybrid/__init__.py -------------------------------------------------------------------------------- /xfuser/core/long_ctx_attention/hybrid/attn_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/core/long_ctx_attention/hybrid/attn_layer.py -------------------------------------------------------------------------------- /xfuser/core/long_ctx_attention/ring/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/core/long_ctx_attention/ring/__init__.py -------------------------------------------------------------------------------- /xfuser/core/long_ctx_attention/ring/ring_flash_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/core/long_ctx_attention/ring/ring_flash_attn.py -------------------------------------------------------------------------------- /xfuser/core/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/core/utils/__init__.py -------------------------------------------------------------------------------- /xfuser/core/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/core/utils/timer.py -------------------------------------------------------------------------------- /xfuser/envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/envs.py -------------------------------------------------------------------------------- /xfuser/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/logger.py -------------------------------------------------------------------------------- /xfuser/model_executor/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /xfuser/model_executor/accelerator/fastcache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/accelerator/fastcache.py -------------------------------------------------------------------------------- /xfuser/model_executor/base_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/base_wrapper.py -------------------------------------------------------------------------------- /xfuser/model_executor/cache/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/cache/__init__.py -------------------------------------------------------------------------------- /xfuser/model_executor/cache/diffusers_adapters/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/cache/diffusers_adapters/__init__.py -------------------------------------------------------------------------------- /xfuser/model_executor/cache/diffusers_adapters/flux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/cache/diffusers_adapters/flux.py -------------------------------------------------------------------------------- /xfuser/model_executor/cache/diffusers_adapters/pixart.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/cache/diffusers_adapters/pixart.py -------------------------------------------------------------------------------- /xfuser/model_executor/cache/diffusers_adapters/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/cache/diffusers_adapters/registry.py -------------------------------------------------------------------------------- /xfuser/model_executor/cache/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/cache/utils.py -------------------------------------------------------------------------------- /xfuser/model_executor/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/layers/__init__.py -------------------------------------------------------------------------------- /xfuser/model_executor/layers/attention_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/layers/attention_processor.py -------------------------------------------------------------------------------- /xfuser/model_executor/layers/base_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/layers/base_layer.py -------------------------------------------------------------------------------- /xfuser/model_executor/layers/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/layers/conv.py -------------------------------------------------------------------------------- /xfuser/model_executor/layers/embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/layers/embeddings.py -------------------------------------------------------------------------------- /xfuser/model_executor/layers/feedforward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/layers/feedforward.py -------------------------------------------------------------------------------- /xfuser/model_executor/layers/register.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/layers/register.py -------------------------------------------------------------------------------- /xfuser/model_executor/layers/usp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/layers/usp.py -------------------------------------------------------------------------------- /xfuser/model_executor/layers/usp_legacy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/layers/usp_legacy.py -------------------------------------------------------------------------------- /xfuser/model_executor/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/models/__init__.py -------------------------------------------------------------------------------- /xfuser/model_executor/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/models/base_model.py -------------------------------------------------------------------------------- /xfuser/model_executor/models/customized/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /xfuser/model_executor/models/customized/step_video_t2v/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /xfuser/model_executor/models/customized/step_video_t2v/attentions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/models/customized/step_video_t2v/attentions.py -------------------------------------------------------------------------------- /xfuser/model_executor/models/customized/step_video_t2v/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/models/customized/step_video_t2v/blocks.py -------------------------------------------------------------------------------- /xfuser/model_executor/models/customized/step_video_t2v/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/models/customized/step_video_t2v/linear.py -------------------------------------------------------------------------------- /xfuser/model_executor/models/customized/step_video_t2v/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/models/customized/step_video_t2v/normalization.py -------------------------------------------------------------------------------- /xfuser/model_executor/models/customized/step_video_t2v/rope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/models/customized/step_video_t2v/rope.py -------------------------------------------------------------------------------- /xfuser/model_executor/models/customized/step_video_t2v/tp_applicator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/models/customized/step_video_t2v/tp_applicator.py -------------------------------------------------------------------------------- /xfuser/model_executor/models/transformers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/models/transformers/__init__.py -------------------------------------------------------------------------------- /xfuser/model_executor/models/transformers/base_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/models/transformers/base_transformer.py -------------------------------------------------------------------------------- /xfuser/model_executor/models/transformers/cogvideox_transformer_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/models/transformers/cogvideox_transformer_3d.py -------------------------------------------------------------------------------- /xfuser/model_executor/models/transformers/consisid_transformer_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/models/transformers/consisid_transformer_3d.py -------------------------------------------------------------------------------- /xfuser/model_executor/models/transformers/hunyuan_transformer_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/models/transformers/hunyuan_transformer_2d.py -------------------------------------------------------------------------------- /xfuser/model_executor/models/transformers/latte_transformer_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/models/transformers/latte_transformer_3d.py -------------------------------------------------------------------------------- /xfuser/model_executor/models/transformers/pixart_transformer_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/models/transformers/pixart_transformer_2d.py -------------------------------------------------------------------------------- /xfuser/model_executor/models/transformers/register.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/models/transformers/register.py -------------------------------------------------------------------------------- /xfuser/model_executor/models/transformers/transformer_flux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/models/transformers/transformer_flux.py -------------------------------------------------------------------------------- /xfuser/model_executor/models/transformers/transformer_sd3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/models/transformers/transformer_sd3.py -------------------------------------------------------------------------------- /xfuser/model_executor/patch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /xfuser/model_executor/patch/unet_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/patch/unet_patch.py -------------------------------------------------------------------------------- /xfuser/model_executor/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/pipelines/__init__.py -------------------------------------------------------------------------------- /xfuser/model_executor/pipelines/base_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/pipelines/base_pipeline.py -------------------------------------------------------------------------------- /xfuser/model_executor/pipelines/dit_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/pipelines/dit_pipeline.py -------------------------------------------------------------------------------- /xfuser/model_executor/pipelines/fastcache_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/pipelines/fastcache_pipeline.py -------------------------------------------------------------------------------- /xfuser/model_executor/pipelines/pipeline_cogvideox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/pipelines/pipeline_cogvideox.py -------------------------------------------------------------------------------- /xfuser/model_executor/pipelines/pipeline_consisid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/pipelines/pipeline_consisid.py -------------------------------------------------------------------------------- /xfuser/model_executor/pipelines/pipeline_flux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/pipelines/pipeline_flux.py -------------------------------------------------------------------------------- /xfuser/model_executor/pipelines/pipeline_hunyuandit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/pipelines/pipeline_hunyuandit.py -------------------------------------------------------------------------------- /xfuser/model_executor/pipelines/pipeline_latte.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/pipelines/pipeline_latte.py -------------------------------------------------------------------------------- /xfuser/model_executor/pipelines/pipeline_pixart_alpha.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/pipelines/pipeline_pixart_alpha.py -------------------------------------------------------------------------------- /xfuser/model_executor/pipelines/pipeline_pixart_sigma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/pipelines/pipeline_pixart_sigma.py -------------------------------------------------------------------------------- /xfuser/model_executor/pipelines/pipeline_stable_diffusion_3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/pipelines/pipeline_stable_diffusion_3.py -------------------------------------------------------------------------------- /xfuser/model_executor/pipelines/pipeline_stable_diffusion_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/pipelines/pipeline_stable_diffusion_xl.py -------------------------------------------------------------------------------- /xfuser/model_executor/pipelines/register.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/pipelines/register.py -------------------------------------------------------------------------------- /xfuser/model_executor/schedulers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/schedulers/__init__.py -------------------------------------------------------------------------------- /xfuser/model_executor/schedulers/base_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/schedulers/base_scheduler.py -------------------------------------------------------------------------------- /xfuser/model_executor/schedulers/register.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/schedulers/register.py -------------------------------------------------------------------------------- /xfuser/model_executor/schedulers/scheduling_ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/schedulers/scheduling_ddim.py -------------------------------------------------------------------------------- /xfuser/model_executor/schedulers/scheduling_ddim_cogvideox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/schedulers/scheduling_ddim_cogvideox.py -------------------------------------------------------------------------------- /xfuser/model_executor/schedulers/scheduling_ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/schedulers/scheduling_ddpm.py -------------------------------------------------------------------------------- /xfuser/model_executor/schedulers/scheduling_dpm_cogvideox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/schedulers/scheduling_dpm_cogvideox.py -------------------------------------------------------------------------------- /xfuser/model_executor/schedulers/scheduling_dpmsolver_multistep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/schedulers/scheduling_dpmsolver_multistep.py -------------------------------------------------------------------------------- /xfuser/model_executor/schedulers/scheduling_flow_match_euler_discrete.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/model_executor/schedulers/scheduling_flow_match_euler_discrete.py -------------------------------------------------------------------------------- /xfuser/parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/parallel.py -------------------------------------------------------------------------------- /xfuser/ray/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /xfuser/ray/pipeline/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /xfuser/ray/pipeline/base_executor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/ray/pipeline/base_executor.py -------------------------------------------------------------------------------- /xfuser/ray/pipeline/pipeline_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/ray/pipeline/pipeline_utils.py -------------------------------------------------------------------------------- /xfuser/ray/pipeline/ray_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/ray/pipeline/ray_utils.py -------------------------------------------------------------------------------- /xfuser/ray/worker/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /xfuser/ray/worker/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/ray/worker/utils.py -------------------------------------------------------------------------------- /xfuser/ray/worker/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/ray/worker/worker.py -------------------------------------------------------------------------------- /xfuser/ray/worker/worker_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NoakLiu/FastCache-xDiT/HEAD/xfuser/ray/worker/worker_wrappers.py --------------------------------------------------------------------------------