lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-29 10:41:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update unet.py

Browse Source
This commit is contained in:
layerdiffusion
2024-08-02 04:22:05 -07:00
parent 40476bb8b3
commit 2104e12dab
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
backend/nn/unet.py
View File

@@ -281,10 +281,7 @@ class BasicTransformerBlock(nn.Module):
if self.attn2 is not None: if self.attn2 is not None:
n = self.norm2(x) n = self.norm2(x)
if self.switch_temporal_ca_to_sa: context_attn2 = context
context_attn2 = n
else:
context_attn2 = context
value_attn2 = None value_attn2 = None
if "attn2_patch" in transformer_patches: if "attn2_patch" in transformer_patches:
patch = transformer_patches["attn2_patch"] patch = transformer_patches["attn2_patch"]