Merge branch 'main' of hf.co:spaces/xilluill/KV-Edit into main

.gitattributes

@@ -32,4 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.xz filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,2 @@
+__pycache__/
+result/

README.md

@@ -1,13 +1,14 @@
 ---
-title: KV Edit
-emoji: 🐢
-colorFrom: gray
+title: KV-Edit
+emoji: 🎨
+colorFrom: blue
 colorTo: pink
 sdk: gradio
 sdk_version: 5.16.2
 app_file: app.py
 pinned: false
 license: apache-2.0
+short_description: Precise Background Preservation in Editing
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -194,9 +194,9 @@ def create_demo(model_name: str):
         
         with gr.Row():
             with gr.Column():
-                source_prompt = gr.Textbox(label="Source Prompt", value='In a cluttered wooden cabin, a workbench holds a green neon sign that reads "I love here"' )
+                source_prompt = gr.Textbox(label="Source Prompt", value='' )
                 inversion_num_steps = gr.Slider(1, 50, 28, step=1, label="Number of inversion steps")
-                target_prompt = gr.Textbox(label="Target Prompt", value='In a cluttered wooden cabin, a workbench holds a green neon sign that reads "I love iccv"' )
+                target_prompt = gr.Textbox(label="Target Prompt", value='' )
                 denoise_num_steps = gr.Slider(1, 50, 28, step=1, label="Number of denoise steps")
                 brush_canvas = gr.ImageEditor(label="Brush Canvas",
                                                 sources=('upload'), 

flux/model.py

@@ -4,8 +4,8 @@ import torch
 from torch import Tensor, nn
 
 from flux.modules.layers import (DoubleStreamBlock, EmbedND, LastLayer,
-                                 MLPEmbedder, SingleStreamBlock,DoubleStreamBlock_rf,SingleStreamBlock_rf,
-                                    SingleStreamBlock_kv,DoubleStreamBlock_kv,
+                                 MLPEmbedder, SingleStreamBlock,
+                                SingleStreamBlock_kv,DoubleStreamBlock_kv,
                                  timestep_embedding)