Tiled Diffusion & VAE for ComfyUI

See this for more information.

This extension enables large image drawing & upscaling with limited VRAM via the following techniques:

1. Two SOTA diffusion tiling algorithms: Mixture of Diffusers and MultiDiffusion
2. pkuliyi2015 & Kahsolt's Tiled VAE algorithm.
3. pkuliyi2015 & Kahsolt's TIled Noise Inversion for better upscaling.

[!NOTE]
Sizes/dimensions are in pixels and then converted to latent-space sizes.

Features

• SDXL model support
• ControlNet support
• StableSR support
• Tiled Noise Inversion
• Tiled VAE
• Regional Prompt Control
• Img2img upscale
• Ultra-Large image generation

Some conditioning nodes like SetArea or GLIGEN aren't working at the moment.

Tiled Diffusion

[!TIP]
Set tile_overlap to 0 and denoise to 1 to see the tile seams and then adjust the options to your needs. Also, increase tile_batch_size to increase speed (if your machine can handle it).

How can I specify the tiles' arrangement?

If you have the Math Expression node (or something similar), you can use that to pass in the latent that's passed in your KSampler and divide the tile_height/tile_width by the number of rows/columns you want.

C = number of columns you want
R = number of rows you want

pixel width of input image or latent // C = tile_width
pixel height of input image or latent // R = tile_height

Tiled VAE

The recommended tile sizes are given upon the creation of the node based on the available VRAM.

[!NOTE]
Enabling fast for the decoder may produce images with slightly higher contrast and brightness.

Workflows

The following images can be loaded in ComfyUI.

Simple upscale.

4x upscale. 3 passes.

Citation

@article{jimenez2023mixtureofdiffusers,
  title={Mixture of Diffusers for scene composition and high resolution image generation},
  author={Álvaro Barbero Jiménez},
  journal={arXiv preprint arXiv:2302.02412},
  year={2023}
}

@article{bar2023multidiffusion,
  title={MultiDiffusion: Fusing Diffusion Paths for Controlled Image Generation},
  author={Bar-Tal, Omer and Yariv, Lior and Lipman, Yaron and Dekel, Tali},
  journal={arXiv preprint arXiv:2302.08113},
  year={2023}
}