Update optimization.py (#3)

- Update optimization.py (8dc6c106b60acd3e7a1089f80e31f02d7a789c3a)
- Update app.py (de0bec3971c17324a1f71c8e40b8e9da28546c6d)

app.py

@@ -299,7 +299,7 @@ def update_history(new_image, history):
 def use_history_as_input(evt: gr.SelectData, history):
     """Sets the selected history image as the new input image."""
     if history and evt.index < len(history):
-        return gr.update(value=history[evt.index])
+        return gr.update(value=history[evt.index][0])
     return gr.update()
 
 def use_output_as_input(output_image):
@@ -441,16 +441,7 @@ with gr.Blocks(css=css) as demo:
             <img src="https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen-Image/qwen_image_edit_logo.png" alt="Qwen-Image Edit Logo" width="400" style="display: block; margin: 0 auto;">
         </div>
         """)
-        gr.Markdown("""
-        ## Qwen-Image Edit with Outpainting
-        
-        Extend your images beyond their original boundaries with intelligent outpainting. The model will generate new content that seamlessly blends with your original image.
-        
-        **Tips:**
-        - Use the preview button to see which areas will be generated before running
-        - Click on any image in the history to use it as a new input
-        - Try different alignments to expand your image in specific directions
-        
+        gr.Markdown("""        
         [Learn more](https://github.com/QwenLM/Qwen-Image) about the Qwen-Image series. 
         Try on [Qwen Chat](https://chat.qwen.ai/), or [download model](https://huggingface.co/Qwen/Qwen-Image-Edit) to run locally.
         """)
@@ -478,9 +469,9 @@ with gr.Blocks(css=css) as demo:
                         label="Alignment"
                     )
                 
-                run_button = gr.Button("🎨 Outpaint!", variant="primary")
+                run_button = gr.Button("run", variant="primary")
                 
-                with gr.Accordion("⚙️ Outpainting Settings", open=False) as settings_panel:
+                with gr.Accordion("Outpainting Settings", open=False) as settings_panel:
                     with gr.Row():
                         width_slider = gr.Slider(
                             label="Target Width",
@@ -555,9 +546,9 @@ with gr.Blocks(css=css) as demo:
                         num_inference_steps = gr.Slider(
                             label="Number of inference steps",
                             minimum=1,
-                            maximum=50,
+                            maximum=28,
                             step=1,
-                            value=50,
+                            value=8,
                         )
 
                     rewrite_prompt = gr.Checkbox(

optimization.py

@@ -49,6 +49,13 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
 
     @spaces.GPU(duration=1500)
     def compile_transformer():
+        
+        pipeline.load_lora_weights(
+        "lightx2v/Qwen-Image-Lightning", 
+        weight_name="Qwen-Image-Lightning-8steps-V1.1.safetensors"
+        )
+        pipeline.fuse_lora()
+        pipeline.unload_lora_weights()
 
         with spaces.aoti_capture(pipeline.transformer) as call:
             pipeline(*args, **kwargs)