From b0cb57ee235adc688e127b888993df680c22ebfa Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Sat, 16 Mar 2024 11:23:56 -0400 Subject: [PATCH] add error handler around fused call --- modules/sd_models.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/modules/sd_models.py b/modules/sd_models.py index 91d793f82..9f578ee93 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -727,8 +727,11 @@ def set_diffuser_options(sd_model, vae = None, op: str = 'model'): set_diffusers_attention(sd_model, DynamicAttnProcessorSDP()) if shared.opts.diffusers_fuse_projections and hasattr(sd_model, 'fuse_qkv_projections'): - shared.log.debug(f'Setting {op}: enable fused projections') - sd_model.fuse_qkv_projections() + try: + sd_model.fuse_qkv_projections() + shared.log.debug(f'Setting {op}: enable fused projections') + except Exception as e: + shared.log.error(f'Error enabling fused projections: {e}') if shared.opts.diffusers_eval: if hasattr(sd_model, "unet") and hasattr(sd_model.unet, "requires_grad_"): sd_model.unet.requires_grad_(False)