From 6847d0de19dd0fc419b010fa247435cad2fe7f86 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Sat, 9 Mar 2024 09:30:20 -0500 Subject: [PATCH] fix hires with control --- modules/processing_diffusers.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/modules/processing_diffusers.py b/modules/processing_diffusers.py index 8672a9b4e..6ade59e8f 100644 --- a/modules/processing_diffusers.py +++ b/modules/processing_diffusers.py @@ -442,8 +442,10 @@ def process_diffusers(p: processing.StableDiffusionProcessing): shared.log.error(f'Processing: args={base_args} {e}') errors.display(e, 'Processing') - if hasattr(shared.sd_model, 'embedding_db') and len(shared.sd_model.embedding_db.embeddings_used) > 0: + if hasattr(shared.sd_model, 'embedding_db') and len(shared.sd_model.embedding_db.embeddings_used) > 0: # register used embeddings p.extra_generation_params['Embeddings'] = ', '.join(shared.sd_model.embedding_db.embeddings_used) + if hasattr(p, 'task_args') and p.task_args.get('image', None) is not None: # replace input with output so it can be used by hires/refine + p.task_args['image'] = output.images shared.state.nextjob() if shared.state.interrupted or shared.state.skipped: