update refiner start/end

This commit is contained in:
Vladimir Mandic
2023-07-26 20:49:24 -04:00
parent 0bc389594c
commit 3437132817
7 changed files with 29 additions and 36 deletions
+2 -3
View File
@@ -139,7 +139,7 @@ def process_diffusers(p: StableDiffusionProcessing, seeds, prompts, negative_pro
negative_prompt_2=[p.refiner_negative] if len(p.refiner_negative) > 0 else negative_prompts,
eta=shared.opts.eta_ddim,
guidance_rescale=p.diffusers_guidance_rescale,
denoising_end=p.refiner_denoise_start if refiner_enabled else None,
denoising_end=p.refiner_start if p.refiner_start > 0 and p.refiner_start < 1 else None,
# aesthetic_score=shared.opts.diffusers_aesthetics_score,
output_type='latent' if hasattr(shared.sd_model, 'vae') else 'np',
**task_specific_kwargs
@@ -189,8 +189,7 @@ def process_diffusers(p: StableDiffusionProcessing, seeds, prompts, negative_pro
guidance_scale=p.image_cfg_scale if p.image_cfg_scale is not None else p.cfg_scale,
guidance_rescale=p.diffusers_guidance_rescale,
# aesthetic_score=shared.opts.diffusers_aesthetics_score,
denoising_start=p.refiner_denoise_start,
denoising_end=p.refiner_denoise_end,
denoising_start=p.refiner_start if p.refiner_start > 0 and p.refiner_start < 1 else None,
image=output.images[i],
output_type='latent' if hasattr(shared.sd_refiner, 'vae') else 'np',
)