Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-06 13:00:44 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix base and refiner when using diffusers

Browse Source 
With diffusers 0.19 densoise_start and _end started applying.
This commit is contained in:
Kubuxu
2023-07-27 00:45:43 +01:00
parent 2ee4c88a63
commit fc1a7ad0b1
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_diffusers.py
+3 -3
View File
@@ -130,6 +130,7 @@ def process_diffusers(p: StableDiffusionProcessing, seeds, prompts, negative_pro
if shared.opts.diffusers_move_base:
shared.sd_model.to(devices.device)
refiner_enabled = shared.sd_refiner is not None and p.enable_hr
pipe_args = set_pipeline_args(
model=shared.sd_model,
prompt=prompts,
@@ -138,8 +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_start=p.refiner_denoise_start,
denoising_end=p.refiner_denoise_end,
denoising_end=p.refiner_denoise_start if refiner_enabled else None,
# aesthetic_score=shared.opts.diffusers_aesthetics_score,
output_type='latent' if hasattr(shared.sd_model, 'vae') else 'np',
**task_specific_kwargs
@@ -152,7 +152,7 @@ def process_diffusers(p: StableDiffusionProcessing, seeds, prompts, negative_pro
if shared.sd_refiner is None or not p.enable_hr:
output.images = vae_decode(output.images, shared.sd_model)
if shared.sd_refiner is not None and p.enable_hr:
if refiner_enabled:
for i in range(len(output.images)):
if shared.opts.save and not p.do_not_save_samples and shared.opts.save_images_before_refiner and hasattr(shared.sd_model, 'vae'):
from modules.processing import create_infotext