├── .gitattributes ├── .github └── workflows │ └── publish.yml ├── .gitignore ├── LICENSE ├── README.md ├── __init__.py ├── configs ├── causal_video_vae_config.json ├── miniflux_transformer_config.json └── mmdit_transformer_config.json ├── diffusion_schedulers ├── __init__.py ├── scheduling_cosine_ddpm.py └── scheduling_flow_matching.py ├── examples ├── pyramid_flow_miniflux_768_img2vid_example_01.json ├── pyramid_flow_miniflux_img2vid_example_01.json └── pyramid_flow_miniflux_text2vid_example_01.json ├── fp8_optimization.py ├── latent_preview.py ├── nodes.py ├── pyproject.toml ├── pyramid_dit ├── __init__.py ├── flux_modules │ ├── __init__.py │ ├── modeling_embedding.py │ ├── modeling_flux_block.py │ ├── modeling_normalization.py │ └── modeling_pyramid_flux.py ├── mmdit_modules │ ├── __init__.py │ ├── modeling_embedding.py │ ├── modeling_mmdit_block.py │ ├── modeling_normalization.py │ ├── modeling_pyramid_mmdit.py │ └── modeling_text_encoder.py └── pyramid_dit_for_video_gen_pipeline.py ├── requirements.txt ├── utils.py └── video_vae ├── __init__.py ├── context_parallel_ops.py ├── modeling_block.py ├── modeling_causal_conv.py ├── modeling_causal_vae.py ├── modeling_discriminator.py ├── modeling_enc_dec.py ├── modeling_loss.py ├── modeling_lpips.py └── modeling_resnet.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/.gitattributes -------------------------------------------------------------------------------- /.github/workflows/publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/.github/workflows/publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/__init__.py -------------------------------------------------------------------------------- /configs/causal_video_vae_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/configs/causal_video_vae_config.json -------------------------------------------------------------------------------- /configs/miniflux_transformer_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/configs/miniflux_transformer_config.json -------------------------------------------------------------------------------- /configs/mmdit_transformer_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/configs/mmdit_transformer_config.json -------------------------------------------------------------------------------- /diffusion_schedulers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/diffusion_schedulers/__init__.py -------------------------------------------------------------------------------- /diffusion_schedulers/scheduling_cosine_ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/diffusion_schedulers/scheduling_cosine_ddpm.py -------------------------------------------------------------------------------- /diffusion_schedulers/scheduling_flow_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/diffusion_schedulers/scheduling_flow_matching.py -------------------------------------------------------------------------------- /examples/pyramid_flow_miniflux_768_img2vid_example_01.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/examples/pyramid_flow_miniflux_768_img2vid_example_01.json -------------------------------------------------------------------------------- /examples/pyramid_flow_miniflux_img2vid_example_01.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/examples/pyramid_flow_miniflux_img2vid_example_01.json -------------------------------------------------------------------------------- /examples/pyramid_flow_miniflux_text2vid_example_01.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/examples/pyramid_flow_miniflux_text2vid_example_01.json -------------------------------------------------------------------------------- /fp8_optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/fp8_optimization.py -------------------------------------------------------------------------------- /latent_preview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/latent_preview.py -------------------------------------------------------------------------------- /nodes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/nodes.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/pyproject.toml -------------------------------------------------------------------------------- /pyramid_dit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/pyramid_dit/__init__.py -------------------------------------------------------------------------------- /pyramid_dit/flux_modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/pyramid_dit/flux_modules/__init__.py -------------------------------------------------------------------------------- /pyramid_dit/flux_modules/modeling_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/pyramid_dit/flux_modules/modeling_embedding.py -------------------------------------------------------------------------------- /pyramid_dit/flux_modules/modeling_flux_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/pyramid_dit/flux_modules/modeling_flux_block.py -------------------------------------------------------------------------------- /pyramid_dit/flux_modules/modeling_normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/pyramid_dit/flux_modules/modeling_normalization.py -------------------------------------------------------------------------------- /pyramid_dit/flux_modules/modeling_pyramid_flux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/pyramid_dit/flux_modules/modeling_pyramid_flux.py -------------------------------------------------------------------------------- /pyramid_dit/mmdit_modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/pyramid_dit/mmdit_modules/__init__.py -------------------------------------------------------------------------------- /pyramid_dit/mmdit_modules/modeling_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/pyramid_dit/mmdit_modules/modeling_embedding.py -------------------------------------------------------------------------------- /pyramid_dit/mmdit_modules/modeling_mmdit_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/pyramid_dit/mmdit_modules/modeling_mmdit_block.py -------------------------------------------------------------------------------- /pyramid_dit/mmdit_modules/modeling_normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/pyramid_dit/mmdit_modules/modeling_normalization.py -------------------------------------------------------------------------------- /pyramid_dit/mmdit_modules/modeling_pyramid_mmdit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/pyramid_dit/mmdit_modules/modeling_pyramid_mmdit.py -------------------------------------------------------------------------------- /pyramid_dit/mmdit_modules/modeling_text_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/pyramid_dit/mmdit_modules/modeling_text_encoder.py -------------------------------------------------------------------------------- /pyramid_dit/pyramid_dit_for_video_gen_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/pyramid_dit/pyramid_dit_for_video_gen_pipeline.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/utils.py -------------------------------------------------------------------------------- /video_vae/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/video_vae/__init__.py -------------------------------------------------------------------------------- /video_vae/context_parallel_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/video_vae/context_parallel_ops.py -------------------------------------------------------------------------------- /video_vae/modeling_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/video_vae/modeling_block.py -------------------------------------------------------------------------------- /video_vae/modeling_causal_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/video_vae/modeling_causal_conv.py -------------------------------------------------------------------------------- /video_vae/modeling_causal_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/video_vae/modeling_causal_vae.py -------------------------------------------------------------------------------- /video_vae/modeling_discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/video_vae/modeling_discriminator.py -------------------------------------------------------------------------------- /video_vae/modeling_enc_dec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/video_vae/modeling_enc_dec.py -------------------------------------------------------------------------------- /video_vae/modeling_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/video_vae/modeling_loss.py -------------------------------------------------------------------------------- /video_vae/modeling_lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/video_vae/modeling_lpips.py -------------------------------------------------------------------------------- /video_vae/modeling_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-PyramidFlowWrapper/HEAD/video_vae/modeling_resnet.py --------------------------------------------------------------------------------