lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-05-01 11:41:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update diffusers_patcher.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-28 10:23:29 -08:00
parent 8ec474c75b
commit 2c64478e10
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modules_forge/diffusers_patcher.py
View File

@@ -23,6 +23,7 @@ class DiffusersModelPatcher:
self.pipeline = pipeline_class.from_pretrained(*args, **kwargs)
if hasattr(self.pipeline, 'unet'):
if hasattr(self.pipeline.unet, 'set_attn_processor'):
self.pipeline.unet.set_attn_processor(AttnProcessor2_0())
print('Attention optimization applied to DiffusersModelPatcher')