diff options
| author | Elias Sundqvist <github.elias@sqvist.com> | 2022-09-16 06:40:43 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-09-16 10:19:13 +0300 |
| commit | a441cd563aa04c8ebfc4d02a4a004f649ad44f71 (patch) | |
| tree | 5f6d29bbcd19798e8bfeb00434f43d7f8943f089 | |
| parent | 2aec11d263e98580787bc3f3130a09ab2d1cdfc9 (diff) | |
Reduce code duplication
| -rw-r--r-- | scripts/img2imgalt.py | 20 |
1 files changed, 4 insertions, 16 deletions
diff --git a/scripts/img2imgalt.py b/scripts/img2imgalt.py index 6581eaad..dbda3255 100644 --- a/scripts/img2imgalt.py +++ b/scripts/img2imgalt.py @@ -105,25 +105,13 @@ class Script(scripts.Script): sampler = samplers[p.sampler_index].constructor(p.sd_model)
sigmas = sampler.model_wrap.get_sigmas(p.steps)
-
- t_enc = int(min(p.denoising_strength, 0.999) * p.steps)
noise_dt = combined_noise - ( p.init_latent / sigmas[0] )
- noise_dt = noise_dt * sigmas[p.steps - t_enc - 1]
-
- noise = p.init_latent + noise_dt
-
- sigma_sched = sigmas[p.steps - t_enc - 1:]
-
- sampler.model_wrap_cfg.mask = p.mask
- sampler.model_wrap_cfg.nmask = p.nmask
- sampler.model_wrap_cfg.init_latent = p.init_latent
-
- if hasattr(K.sampling, 'trange'):
- K.sampling.trange = lambda *args, **kwargs: sd_samplers.extended_trange(*args, **kwargs)
-
+
p.seed = p.seed + 1
- return sampler.func(sampler.model_wrap_cfg, noise, sigma_sched, extra_args={'cond': conditioning, 'uncond': unconditional_conditioning, 'cond_scale': p.cfg_scale}, disable=False, callback=sampler.callback_state)
+
+ return sampler.sample_img2img(p, p.init_latent, noise_dt, conditioning, unconditional_conditioning)
+
p.sample = sample_extra
|