Block swapping is supported for Wan, HunyuanVideo, Flux, and Chroma. Big thanks to @kohya-ss and Musubi Tuner from which most of the implementation is taken. See the example hunyuan_video.toml file ...

Thanks to @zjysteven for his efforts. Below shows an example with stable-diffusion-2-1-base. import torch from PIL import Image, ImageDraw from copy import deepcopy from ...