fix controlnet with batch count

Signed-off-by: Vladimir Mandic <[email protected]>

CHANGELOG.md

@@ -1,6 +1,6 @@
 # Change Log for SD.Next
 
-## Update for 2025-01-04
+## Update for 2025-01-05
 
 - [Allegro Video](https://huggingface.co/rhymes-ai/Allegro)  
   - optimizations: full offload and quantization support  
@@ -20,9 +20,8 @@
   - add explicit detailer steps setting  
 - **SysInfo**:
   - update to collected data and benchmarks  
-- **Progress**:
+- **Refactor**:
   - refactored progress monitoring, job updates and live preview  
-- **Metadata**:
   - improved metadata save and restore  
 - **Fixes**:
   - explict clear caches on model load  
@@ -34,6 +33,7 @@
   - samplers test for scale noise before using  
   - scheduler api  
   - controlnet with hires  
+  - controlnet with batch count  
 
 ## Update for 2024-12-31
 

modules/processing_diffusers.py

@@ -201,7 +201,7 @@ def process_hires(p: processing.StableDiffusionProcessing, output):
             update_sampler(p, shared.sd_model, second_pass=True)
             orig_denoise = p.denoising_strength
             p.denoising_strength = strength
-            p.task_args.pop('image', None) # remove image override from hires
+            orig_image = p.task_args.pop('image', None) # remove image override from hires
             hires_args = set_pipeline_args(
                 p=p,
                 model=shared.sd_model,
@@ -245,6 +245,8 @@ def process_hires(p: processing.StableDiffusionProcessing, output):
                 shared.log.error(f'Processing step=hires: args={hires_args} {e}')
                 errors.display(e, 'Processing')
                 modelstats.analyze()
+            if orig_image is not None:
+                p.task_args['image'] = orig_image
             p.denoising_strength = orig_denoise
         shared.state.job = prev_job
         shared.state.nextjob()