diff --git a/CHANGELOG.md b/CHANGELOG.md index a606bfb16..3e1ab23fe 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -22,6 +22,7 @@ Another big one, but now improvements to both **diffusers** and **original** bac - other: - major refactoring of the javascript code includes fixes for text selections and navigation + - system info tab now reports on nvidia driver version as well - minor fixes in extra-networks - installer handles origin changes for submodules diff --git a/extensions-builtin/sd-extension-system-info b/extensions-builtin/sd-extension-system-info index 29b6f1602..4f0e81050 160000 --- a/extensions-builtin/sd-extension-system-info +++ b/extensions-builtin/sd-extension-system-info @@ -1 +1 @@ -Subproject commit 29b6f1602366556562bb76644444db2a51bd60bb +Subproject commit 4f0e81050f1e429a2531d0bb702029f5cf18deab diff --git a/modules/processing.py b/modules/processing.py index 5ed798623..b9ceacab0 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -761,6 +761,8 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed: model=shared.sd_refiner, prompt=[p.refiner_prompt] if len(p.refiner_prompt) > 0 else prompts, negative_prompt=[p.refiner_negative] if len(p.refiner_negative) > 0 else negative_prompts, + num_inference_steps=p.refiner_steps, + denoising_start=p.refiner_denoise, image=output.images[0], output_type='np' ) diff --git a/modules/sd_models.py b/modules/sd_models.py index f78cdf5fb..17986ee58 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -382,7 +382,10 @@ def load_model_weights(model: torch.nn.Module, checkpoint_info: CheckpointInfo, shared.opts.data["sd_model_checkpoint"] = checkpoint_info.title if state_dict is None: state_dict = get_checkpoint_state_dict(checkpoint_info, timer) - model.load_state_dict(state_dict, strict=False) + try: + model.load_state_dict(state_dict, strict=False) + except Exception as e: + shared.log.error(f'Error loading model weights: {checkpoint_info.filename} {e}') del state_dict timer.record("apply") if shared.opts.sd_checkpoint_cache > 0: