Commit c5d4c87c authored by invincibledude's avatar invincibledude

hr conditioning

parent 4e0cf7d4
...@@ -635,10 +635,11 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed: ...@@ -635,10 +635,11 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
with devices.autocast(): with devices.autocast():
if type(p) == StableDiffusionProcessingTxt2Img: if type(p) == StableDiffusionProcessingTxt2Img:
if p.enable_hr: 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, hr_conditioning=hr_c, hr_unconditional_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, else:
subseeds=subseeds, samples_ddim = p.sample(conditioning=c, unconditional_conditioning=uc, seeds=seeds, subseeds=subseeds,
subseed_strength=p.subseed_strength, prompts=prompts) subseed_strength=p.subseed_strength, prompts=prompts)
else: else:
samples_ddim = p.sample(conditioning=c, unconditional_conditioning=uc, seeds=seeds, samples_ddim = p.sample(conditioning=c, unconditional_conditioning=uc, seeds=seeds,
subseeds=subseeds, subseeds=subseeds,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment