From 36cb64eef63612a111236a336f6b625162c526d8 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Sun, 4 Feb 2024 02:14:09 +0300 Subject: [PATCH] OpenVINO change recompile order --- modules/processing_diffusers.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/modules/processing_diffusers.py b/modules/processing_diffusers.py index f047ed0c6..abfe3a3dc 100644 --- a/modules/processing_diffusers.py +++ b/modules/processing_diffusers.py @@ -373,6 +373,9 @@ def process_diffusers(p: processing.StableDiffusionProcessing): if shared.opts.diffusers_move_base and not getattr(shared.sd_model, 'has_accelerate', False): shared.sd_model.to(devices.device) + # recompile if a paramater chages + recompile_model() + # pipeline type is set earlier in processing, but check for sanity is_control = getattr(p, 'is_control', False) is True has_images = len(getattr(p, 'init_images' ,[])) > 0 @@ -403,7 +406,6 @@ def process_diffusers(p: processing.StableDiffusionProcessing): clip_skip=p.clip_skip, desc='Base', ) - recompile_model() update_sampler(shared.sd_model) shared.state.sampling_steps = base_args['num_inference_steps'] p.extra_generation_params['Pipeline'] = shared.sd_model.__class__.__name__ @@ -462,10 +464,10 @@ def process_diffusers(p: processing.StableDiffusionProcessing): output.images = resize_hires(p, latents=output.images) if (latent_scale_mode is not None or p.hr_force) and p.denoising_strength > 0: p.ops.append('hires') + recompile_model(hires=True) shared.sd_model = sd_models.set_diffuser_pipe(shared.sd_model, sd_models.DiffusersTaskType.IMAGE_2_IMAGE) if shared.sd_model.__class__.__name__ == "OnnxRawPipeline": shared.sd_model = preprocess_onnx_pipeline(p, is_refiner_enabled()) - recompile_model(hires=True) update_sampler(shared.sd_model, second_pass=True) hires_args = set_pipeline_args( model=shared.sd_model,