Вот код
Если хотите поподробнее (с инструкцией как запустить), то пишите в комментариях.
P.S. ComfyUI использую, тратится 1 секунда на 1 изображение.