From ad778e688ad1e77e55622b9c556b616e2cbfed7d Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 15:00:03 -0800 Subject: [PATCH] Update processing.py --- modules/processing.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/modules/processing.py b/modules/processing.py index b2f85793..2b5723af 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -834,6 +834,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed: sd_models.reload_model_weights() # model can be changed for example by refiner + p.sd_model.forge_objects = p.sd_model.forge_objects_original.shallow_copy() p.prompts = p.all_prompts[n * p.batch_size:(n + 1) * p.batch_size] p.negative_prompts = p.all_negative_prompts[n * p.batch_size:(n + 1) * p.batch_size] p.seeds = p.all_seeds[n * p.batch_size:(n + 1) * p.batch_size] @@ -852,6 +853,8 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed: if not p.disable_extra_networks: extra_networks.activate(p, p.extra_network_data) + p.sd_model.forge_objects = p.sd_model.forge_objects_after_applying_lora.shallow_copy() + if p.scripts is not None: p.scripts.process_batch(p, batch_number=n, prompts=p.prompts, seeds=p.seeds, subseeds=p.subseeds)