Думается, Вам стоит обратить внимание на checkpoints, LoRA и VAE для Stable Diffusion.
Добавлено (13 Июня 2025, 00:50)
---------------------------------------------
Например, Вы можете обучить LoRA и применить ноды ControlNet или IPAdepter в ComfyUI. Они изменят стиль загруженного/сгенерированного изображения в рамках моделей которые Вы обучили или изображения-референса которое выбрали.
Пример моего workflow ComfyUI с IPAdapter. Используется готовая модель и LoRA под стилистику мультфильмов и референс.
Моделели для ControlNet ставятся в папку:
Код
./comfyUI/models/controlnet
Модели для IPAdapter ставятся в папку:
Код
./comfyUI/models/clip_vision
и
Код
./comfyUI/models/ipadapter
На github указано какие куда. У меня папки ipadapter не было
Если на Ubuntu 22.04 возникнет ошибка при установке ControlNet OpenPose: ImportError: libGL.so.1: cannot open shared object file: No such file or directory
Код
sudo apt-get update
sudo apt-get install libgl1
sudo reboot
Добавлено (13 Июня 2025, 10:40)
---------------------------------------------
Если нативно для PyTorch, вот эти скрипты может быть...
https://github.com/crowsonkb/style-transfer-pytorch
https://github.com/gordicaleksa/pytorch-neural-style-transfer
