hr conditioning

This commit is contained in:
invincibledude 2023-01-22 15:17:43 +03:00
parent 4e0cf7d4ed
commit c5d4c87c02
1 changed files with 5 additions and 4 deletions

View File

@ -635,10 +635,11 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
with devices.autocast():
if type(p) == StableDiffusionProcessingTxt2Img:
if p.enable_hr:
samples_ddim = p.sample(conditioning=c, unconditional_conditioning=uc, hr_conditioning=hr_c, hr_uconditional_conditioning=hr_uc, seeds=seeds, subseeds=subseeds, subseed_strength=p.subseed_strength, prompts=prompts)
samples_ddim = p.sample(conditioning=c, unconditional_conditioning=uc, seeds=seeds,
subseeds=subseeds,
subseed_strength=p.subseed_strength, prompts=prompts)
samples_ddim = p.sample(conditioning=c, unconditional_conditioning=uc, hr_conditioning=hr_c, hr_unconditional_conditioning=hr_uc, seeds=seeds, subseeds=subseeds, subseed_strength=p.subseed_strength, prompts=prompts)
else:
samples_ddim = p.sample(conditioning=c, unconditional_conditioning=uc, seeds=seeds, subseeds=subseeds,
subseed_strength=p.subseed_strength, prompts=prompts)
else:
samples_ddim = p.sample(conditioning=c, unconditional_conditioning=uc, seeds=seeds,
subseeds=subseeds,