Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

add error handler around fused call

Browse Source
This commit is contained in:
Vladimir Mandic
2024-03-16 11:23:56 -04:00
parent 48d763693c
commit b0cb57ee23
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_models.py
+5 -2
View File
@@ -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)