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

update diffusers

Browse Source
This commit is contained in:
Vladimir Mandic
2023-07-07 12:48:27 -04:00
parent 1c22722c8c
commit bf84ee8168
3 changed files with 16 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing.py
+10
View File
@@ -711,6 +711,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
shared.state.current_latent = latents
shared.state.set_current_image()
# shared.sd_model.to(devices.device)
output = shared.sd_model( # pylint: disable=not-callable
prompt=prompts,
negative_prompt=negative_prompts,
@@ -723,8 +724,13 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
cross_attention_kwargs=cross_attention_kwargs,
**task_specific_kwargs
)
# if shared.cmd_opts.lowvram or shared.cmd_opts.medvram:
# shared.sd_model.to('cpu')
# devices.torch_gc(force=True)
if shared.sd_refiner is not None:
# shared.sd_refiner.to(devices.device)
init_image = output.images[0]
output = shared.sd_refiner( # pylint: disable=not-callable
prompt=prompts,
@@ -738,6 +744,10 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
cross_attention_kwargs=cross_attention_kwargs,
image=init_image
)
# if shared.cmd_opts.lowvram or shared.cmd_opts.medvram:
# shared.sd_refiner.to('cpu')
# devices.torch_gc(force=True)
x_samples_ddim = output.images