update des

app.py

@@ -218,7 +218,7 @@ def create_demo(model_name: str):
         4️⃣ Click the "Edit" button to generate your edited image! <br>
         
         🔔🔔 [<b>Important</b>] Less skip steps, "re_init" and "attn_mask"  will enhance the editing performance, making the results aligned with your text but may lead to discontinuous images.  <br>
-        We recommend trying to increase "attn_scale" to increase attention between mask and background.<br>
+        If you fail because of these three, we recommend trying to increase "attn_scale" to increase attention between mask and background.<br>
         """
     article = r"""
     If our work is helpful, please help to ⭐ the <a href='https://github.com/Xilluill/KV-Edit' target='_blank'>Github Repo</a>. Thanks! 

flux/modules/layers.py

@@ -300,8 +300,7 @@ class DoubleStreamBlock_kv(DoubleStreamBlock):
             if 'attention_mask' in info:
                 attn = attention(q, k, v, pe=pe,attention_mask=info['attention_mask'])
             else:
-                # attn = attention(q, k, v, pe=pe)
-                attn = attention(q, k, v, pe=pe, pe_q = info['pe_mask'],attention_mask=info['attention_scale'])
+                attn = attention(q, k, v, pe=pe)
     
         # elif feature_k_name in info['feature']:
         else:
@@ -315,7 +314,8 @@ class DoubleStreamBlock_kv(DoubleStreamBlock):
             q = torch.cat((txt_q, img_q), dim=2)
             k = torch.cat((txt_k, source_img_k), dim=2)
             v = torch.cat((txt_v, source_img_v), dim=2)
-            attn = attention(q, k, v, pe=pe, pe_q = info['pe_mask'])
+            # attn = attention(q, k, v, pe=pe, pe_q = info['pe_mask'])
+            attn = attention(q, k, v, pe=pe, pe_q = info['pe_mask'],attention_mask=info['attention_scale'])
 
         
         txt_attn, img_attn = attn[:, : txt.shape[1]], attn[:, txt.shape[1] :]