conditional imports and summary timer

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-11-29 10:05:06 -05:00
parent 6aa7a4707e
commit b6963470a9
7 changed files with 31 additions and 15 deletions
+6 -5
View File
@@ -8,8 +8,7 @@ from modules import shared, devices, processing, sd_models, errors, sd_hijack_hy
from modules.processing_helpers import resize_hires, calculate_base_steps, calculate_hires_steps, calculate_refiner_steps, save_intermediate, update_sampler, is_txt2img, is_refiner_enabled
from modules.processing_args import set_pipeline_args
from modules.onnx_impl import preprocess_pipeline as preprocess_onnx_pipeline, check_parameters_changed as olive_check_parameters_changed
from modules.lora.networks import network_load
from modules.lora.networks import timer as network_timer
from modules.lora import networks
debug = shared.log.trace if os.environ.get('SD_DIFFUSERS_DEBUG', None) is not None else lambda *args, **kwargs: None
@@ -427,9 +426,9 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
p.prompts = p.all_prompts[p.iteration * p.batch_size:(p.iteration+1) * p.batch_size]
if p.negative_prompts is None or len(p.negative_prompts) == 0:
p.negative_prompts = p.all_negative_prompts[p.iteration * p.batch_size:(p.iteration+1) * p.batch_size]
network_timer['apply'] = 0
network_timer['restore'] = 0
network_load()
# load loras
networks.network_load()
sd_models.move_model(shared.sd_model, devices.device)
sd_models_compile.openvino_recompile_model(p, hires=False, refiner=False) # recompile if a parameter changes
@@ -459,6 +458,8 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
results = process_decode(p, output)
timer.process.record('decode')
timer.process.add('lora', networks.total_time())
shared.sd_model = orig_pipeline
if p.state == '':
global last_p # pylint: disable=global-statement