Первый ленивый датасет был сгенерен flux 2 klein 4b (и сорс и таргет)
~200 пар 256х256px

После ревью сгенеренного датасета нахлынула ностальгия по шестипалым рукам, благо артефактов в этой заточенной на image edit модельке было полно. Но мы тут не бодишеймингом занимаемся.
Ну и yesflix adaptation c вайтвошингом прилагается.

Первый робкий заход был вдохновлен fastai unet - pretrained resnet18 encoder, 4 levels, bottleneck self-attention, но вместо pixelshuffle icnr на апскейле были обычные upsample conv а-ля openai DDPM unet

Датасет и комфи воркфлоу для генерации с промтами тут
flux 2 klein 4b guide (comfyui)
веса
gguf

Cum ComPyter - Jupyter в ComfyUI

Если вы когда-то писали ноды для Комфи и ловили себя на обмазывании каждой строки принтами, дабы понять, что же там в очередной раз пришло на вход ноды - это для вас.

Ну или если вы устали от хтонического зоопарка из 20 нод, чтобы посчитать паддинг картинки.

Теперь можно все это делать в ноде, делающей вид, что она ячейка в колабе.

Работает как обычная execute python code нода, но со всеми плюшками jupyter.
Можно интерактивно ковыряться с инпутами, строить графики, и все без перезапуска основного воркфлоу. При этом все изменения передаются дальше по графу как обычно.

Лайк, шер, форк, стар

Гит

Мир, если бы pad-unpad divisible by 64 можно было делать в две ноды вместо 10.

ComPyter