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

100% reproduce results from unoptimized codebase

Browse Source
This commit is contained in:
layerdiffusion
2024-08-08 16:40:15 -07:00
parent 9b4922cfca
commit 10e4a3779d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/nn/flux.py
View File

@@ -218,7 +218,7 @@ class DoubleStreamBlock(nn.Module):
del txt_v, img_v
attn = attention(q, k, v, pe=pe)
txt_attn, img_attn = attn[:, :, :txt.shape[1], :], attn[:, :, txt.shape[1]:, :]
txt_attn, img_attn = attn[:, :txt.shape[1]], attn[:, txt.shape[1]:]
del attn
img = img + img_mod1_gate * self.img_attn.proj(img_attn)