File size: 2,164 Bytes
81f4d3a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
# Copyright (C) 2023 Deforum LLC
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, version 3 of the License.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
# Contact the authors: https://deforum.github.io/
from modules.processing import StableDiffusionProcessingImg2Img
from modules.shared import opts, sd_model
import os
def get_webui_sd_pipeline(args, root):
# Set up the pipeline
p = StableDiffusionProcessingImg2Img(
sd_model=sd_model,
outpath_samples = opts.outdir_samples or opts.outdir_img2img_samples,
) # we'll set up the rest later
os.makedirs(args.outdir, exist_ok=True)
p.width, p.height = map(lambda x: x - x % 8, (args.W, args.H))
p.steps = args.steps
p.seed = args.seed
p.sampler_name = args.sampler
p.tiling = args.tiling
p.restore_faces = args.restore_faces
p.subseed = root.subseed
p.subseed_strength = root.subseed_strength
p.seed_resize_from_w = args.seed_resize_from_w
p.seed_resize_from_h = args.seed_resize_from_h
p.fill = args.fill
p.batch_size = 1 # b.size 1 as this is DEFORUM :)
p.seed = args.seed
p.do_not_save_samples = True # Setting this to False will trigger webui's saving mechanism - and we will end up with duplicated files, and another folder within our destination folder - big no no.
p.sampler_name = args.sampler
p.mask_blur = args.mask_overlay_blur
p.extra_generation_params["Mask blur"] = args.mask_overlay_blur
p.n_iter = 1
p.steps = args.steps
p.denoising_strength = 1 - args.strength
p.cfg_scale = args.scale
p.image_cfg_scale = args.pix2pix_img_cfg_scale
p.outpath_samples = args.outdir
return p |