lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-09 01:19:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

marigold_ini

Browse Source
This commit is contained in:
lllyasviel
2024-02-02 00:14:37 -08:00
parent 24db0e241a
commit 7d7094b6ec
11 changed files with 703 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
modules_forge/diffusers_patcher.py
View File

@@ -27,7 +27,11 @@ class DiffusersModelPatcher:
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 = self.pipeline.to(device=offload_device)
if self.dtype == torch.float16:
self.pipeline = self.pipeline.half()
self.pipeline.eval()
self.patcher = ModelPatcher(