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

Update flux.py

Browse Source
This commit is contained in:
layerdiffusion
2024-08-12 21:42:00 -07:00
parent c7aa9c58d2
commit 3589b57ec1
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

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