LICENSE
README.md
pyproject.toml
comfyui/__init__.py
comfyui/video_generator/dit_config.py
comfyui/video_generator/inference_args.py
comfyui/video_generator/load_image.py
comfyui/video_generator/node_helpers.py
comfyui/video_generator/nodes.py
comfyui/video_generator/text_encoder_config.py
comfyui/video_generator/vae_config.py
comfyui/video_generator/video_generator.py
csrc/attn/config_sta.py
csrc/attn/config_vsa.py
csrc/attn/setup_sta.py
csrc/attn/setup_vsa.py
csrc/attn/benchmarks/bench_sta.py
csrc/attn/benchmarks/bench_vsa_hopper.py
csrc/attn/benchmarks/bench_vsa_triton.py
csrc/attn/st_attn/__init__.py
csrc/attn/tests/flex_sta_ref.py
csrc/attn/tests/test_sta.py
csrc/attn/tests/test_vsa.py
csrc/attn/tests/utils.py
csrc/attn/vsa/__init__.py
csrc/attn/vsa/block_sparse_attn_triton.py
csrc/attn/vsa/block_sparse_wrapper.py
csrc/attn/vsa/index.py
demo/gradio_web_demo.py
docs/source/autodoc2_docstring_parser.py
docs/source/conf.py
docs/source/generate_examples.py
examples/inference/basic/basic.py
examples/inference/basic/basic_dmd.py
examples/inference/basic/basic_mps.py
examples/inference/basic/default_args.py
examples/inference/basic/test.py
examples/inference/gradio/gradio_demo.py
examples/inference/lora/wan_lora_inference.py
examples/inference/lora/wan_lora_inference_from_ckpt.py
examples/inference/optimizations/attention_example.py
examples/inference/optimizations/teacache_example.py
examples/inference/sta_mask_search/wan_example.py
fastvideo/STA_configuration.py
fastvideo/__init__.py
fastvideo/envs.py
fastvideo/fastvideo_args.py
fastvideo/forward_context.py
fastvideo/logger.py
fastvideo/utils.py
fastvideo/version.py
fastvideo.egg-info/PKG-INFO
fastvideo.egg-info/SOURCES.txt
fastvideo.egg-info/dependency_links.txt
fastvideo.egg-info/entry_points.txt
fastvideo.egg-info/requires.txt
fastvideo.egg-info/top_level.txt
fastvideo/attention/__init__.py
fastvideo/attention/layer.py
fastvideo/attention/selector.py
fastvideo/attention/backends/__init__.py
fastvideo/attention/backends/abstract.py
fastvideo/attention/backends/flash_attn.py
fastvideo/attention/backends/sage_attn.py
fastvideo/attention/backends/sdpa.py
fastvideo/attention/backends/sliding_tile_attn.py
fastvideo/attention/backends/video_sparse_attn.py
fastvideo/configs/__init__.py
fastvideo/configs/configs.py
fastvideo/configs/utils.py
fastvideo/configs/models/__init__.py
fastvideo/configs/models/base.py
fastvideo/configs/models/dits/__init__.py
fastvideo/configs/models/dits/base.py
fastvideo/configs/models/dits/hunyuanvideo.py
fastvideo/configs/models/dits/stepvideo.py
fastvideo/configs/models/dits/wanvideo.py
fastvideo/configs/models/encoders/__init__.py
fastvideo/configs/models/encoders/base.py
fastvideo/configs/models/encoders/clip.py
fastvideo/configs/models/encoders/llama.py
fastvideo/configs/models/encoders/t5.py
fastvideo/configs/models/vaes/__init__.py
fastvideo/configs/models/vaes/base.py
fastvideo/configs/models/vaes/hunyuanvae.py
fastvideo/configs/models/vaes/stepvideovae.py
fastvideo/configs/models/vaes/wanvae.py
fastvideo/configs/pipelines/__init__.py
fastvideo/configs/pipelines/base.py
fastvideo/configs/pipelines/hunyuan.py
fastvideo/configs/pipelines/registry.py
fastvideo/configs/pipelines/stepvideo.py
fastvideo/configs/pipelines/wan.py
fastvideo/configs/sample/__init__.py
fastvideo/configs/sample/base.py
fastvideo/configs/sample/hunyuan.py
fastvideo/configs/sample/registry.py
fastvideo/configs/sample/stepvideo.py
fastvideo/configs/sample/teacache.py
fastvideo/configs/sample/wan.py
fastvideo/dataset/__init__.py
fastvideo/dataset/latent_datasets.py
fastvideo/dataset/parquet_dataset_iterable_style.py
fastvideo/dataset/parquet_dataset_map_style.py
fastvideo/dataset/preprocessing_datasets.py
fastvideo/dataset/transform.py
fastvideo/dataset/upload_to_hf.py
fastvideo/dataset/utils.py
fastvideo/dataset/validation_dataset.py
fastvideo/dataset/benchmarks/benchmark_parquet_dataset_iterable_style.py
fastvideo/dataset/benchmarks/benchmark_parquet_dataset_map_style.py
fastvideo/dataset/dataloader/schema.py
fastvideo/distributed/__init__.py
fastvideo/distributed/communication_op.py
fastvideo/distributed/parallel_state.py
fastvideo/distributed/utils.py
fastvideo/distributed/device_communicators/__init__.py
fastvideo/distributed/device_communicators/base_device_communicator.py
fastvideo/distributed/device_communicators/cpu_communicator.py
fastvideo/distributed/device_communicators/cuda_communicator.py
fastvideo/distributed/device_communicators/pynccl.py
fastvideo/distributed/device_communicators/pynccl_wrapper.py
fastvideo/entrypoints/__init__.py
fastvideo/entrypoints/video_generator.py
fastvideo/entrypoints/cli/__init__.py
fastvideo/entrypoints/cli/cli_types.py
fastvideo/entrypoints/cli/generate.py
fastvideo/entrypoints/cli/main.py
fastvideo/entrypoints/cli/utils.py
fastvideo/layers/__init__.py
fastvideo/layers/activation.py
fastvideo/layers/custom_op.py
fastvideo/layers/layernorm.py
fastvideo/layers/linear.py
fastvideo/layers/mlp.py
fastvideo/layers/rotary_embedding.py
fastvideo/layers/utils.py
fastvideo/layers/visual_embedding.py
fastvideo/layers/vocab_parallel_embedding.py
fastvideo/layers/lora/linear.py
fastvideo/layers/quantization/__init__.py
fastvideo/layers/quantization/base_config.py
fastvideo/logging_utils/__init__.py
fastvideo/logging_utils/formatter.py
fastvideo/models/__init__.py
fastvideo/models/hf_transformer_utils.py
fastvideo/models/parameter.py
fastvideo/models/registry.py
fastvideo/models/utils.py
fastvideo/models/vision_utils.py
fastvideo/models/dits/base.py
fastvideo/models/dits/hunyuanvideo.py
fastvideo/models/dits/stepvideo.py
fastvideo/models/dits/wanvideo.py
fastvideo/models/encoders/base.py
fastvideo/models/encoders/bert.py
fastvideo/models/encoders/clip.py
fastvideo/models/encoders/llama.py
fastvideo/models/encoders/stepllm.py
fastvideo/models/encoders/t5.py
fastvideo/models/encoders/vision.py
fastvideo/models/loader/__init__.py
fastvideo/models/loader/component_loader.py
fastvideo/models/loader/fsdp_load.py
fastvideo/models/loader/utils.py
fastvideo/models/loader/weight_utils.py
fastvideo/models/schedulers/base.py
fastvideo/models/schedulers/scheduling_flow_match_euler_discrete.py
fastvideo/models/schedulers/scheduling_flow_unipc_multistep.py
fastvideo/models/schedulers/scheduling_unipc_multistep.py
fastvideo/models/vaes/common.py
fastvideo/models/vaes/hunyuanvae.py
fastvideo/models/vaes/stepvideovae.py
fastvideo/models/vaes/wanvae.py
fastvideo/pipelines/__init__.py
fastvideo/pipelines/composed_pipeline_base.py
fastvideo/pipelines/lora_pipeline.py
fastvideo/pipelines/pipeline_batch_info.py
fastvideo/pipelines/pipeline_registry.py
fastvideo/pipelines/basic/__init__.py
fastvideo/pipelines/basic/hunyuan/__init__.py
fastvideo/pipelines/basic/hunyuan/hunyuan_pipeline.py
fastvideo/pipelines/basic/stepvideo/__init__.py
fastvideo/pipelines/basic/stepvideo/stepvideo_pipeline.py
fastvideo/pipelines/basic/wan/__init__.py
fastvideo/pipelines/basic/wan/wan_dmd_pipeline.py
fastvideo/pipelines/basic/wan/wan_i2v_dmd_pipeline.py
fastvideo/pipelines/basic/wan/wan_i2v_pipeline.py
fastvideo/pipelines/basic/wan/wan_pipeline.py
fastvideo/pipelines/preprocess/preprocess_pipeline_base.py
fastvideo/pipelines/preprocess/preprocess_pipeline_i2v.py
fastvideo/pipelines/preprocess/preprocess_pipeline_t2v.py
fastvideo/pipelines/preprocess/v1_preprocess.py
fastvideo/pipelines/stages/__init__.py
fastvideo/pipelines/stages/base.py
fastvideo/pipelines/stages/conditioning.py
fastvideo/pipelines/stages/decoding.py
fastvideo/pipelines/stages/denoising.py
fastvideo/pipelines/stages/encoding.py
fastvideo/pipelines/stages/image_encoding.py
fastvideo/pipelines/stages/input_validation.py
fastvideo/pipelines/stages/latent_preparation.py
fastvideo/pipelines/stages/stepvideo_encoding.py
fastvideo/pipelines/stages/text_encoding.py
fastvideo/pipelines/stages/timestep_preparation.py
fastvideo/pipelines/stages/validators.py
fastvideo/pipelines/training/__init__.py
fastvideo/platforms/__init__.py
fastvideo/platforms/cpu.py
fastvideo/platforms/cuda.py
fastvideo/platforms/interface.py
fastvideo/platforms/mps.py
fastvideo/tests/__init__.py
fastvideo/tests/conftest.py
fastvideo/tests/utils.py
fastvideo/tests/encoders/test_clip_encoder.py
fastvideo/tests/encoders/test_llama_encoder.py
fastvideo/tests/encoders/test_t5_encoder.py
fastvideo/tests/inference/STA/test_STA_inference.py
fastvideo/tests/inference/lora/test_lora_inference_similarity.py
fastvideo/tests/modal/pr_test.py
fastvideo/tests/nightly/test_e2e_i2v_overfit_single_sample.py
fastvideo/tests/nightly/test_e2e_overfit_single_sample.py
fastvideo/tests/old_tests/test_t5_encoder.py
fastvideo/tests/old_tests/test_wan_vae.py
fastvideo/tests/old_tests/tp_example.py
fastvideo/tests/ssim/test_inference_similarity.py
fastvideo/tests/training/VSA/test_training_loss_VSA.py
fastvideo/tests/training/Vanilla/test_training_loss.py
fastvideo/tests/training/distill/test_distill_dmd.py
fastvideo/tests/training/lora/test_lora_training.py
fastvideo/tests/transformers/test_hunyuanvideo.py
fastvideo/tests/transformers/test_hunyuanvideo_load.py
fastvideo/tests/transformers/test_wanvideo.py
fastvideo/tests/vaes/test_hunyuan_vae.py
fastvideo/tests/vaes/test_official_wan_vae.py
fastvideo/tests/vaes/test_wan_vae.py
fastvideo/third_party/__init__.py
fastvideo/third_party/pynvml.py
fastvideo/training/__init__.py
fastvideo/training/activation_checkpoint.py
fastvideo/training/checkpointing_utils.py
fastvideo/training/distillation_pipeline.py
fastvideo/training/training_pipeline.py
fastvideo/training/training_utils.py
fastvideo/training/wan_distillation_pipeline.py
fastvideo/training/wan_i2v_distillation_pipeline.py
fastvideo/training/wan_i2v_training_pipeline.py
fastvideo/training/wan_training_pipeline.py
fastvideo/worker/__init__.py
fastvideo/worker/executor.py
fastvideo/worker/gpu_worker.py
fastvideo/worker/multiproc_executor.py
fastvideo/workflow/workflow_base.py