From 3589b57ec1f59fd7c570b08c85e0108cf3ff67d6 Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Mon, 12 Aug 2024 21:42:00 -0700
Subject: [PATCH] Update flux.py

---
 backend/nn/flux.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/backend/nn/flux.py b/backend/nn/flux.py
index 097e6622..c1514082 100644
--- a/backend/nn/flux.py
+++ b/backend/nn/flux.py
@@ -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