lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-22 07:43:58 +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

5
modules_forge/diffusers_patcher.py
View File

@@ -23,8 +23,9 @@ class DiffusersModelPatcher:
self.pipeline = pipeline_class.from_pretrained(*args, **kwargs)
if hasattr(self.pipeline, 'unet'):
self.pipeline.unet.set_attn_processor(AttnProcessor2_0())
print('Attention optimization applied to DiffusersModelPatcher')
if hasattr(self.pipeline.unet, 'set_attn_processor'):
self.pipeline.unet.set_attn_processor(AttnProcessor2_0())
print('Attention optimization applied to DiffusersModelPatcher')
self.pipeline = self.pipeline.to(device=offload_device, dtype=dtype)
self.pipeline.eval()