Update app.py

app.py

@@ -77,11 +77,9 @@ pipe = Flux2Pipeline.from_pretrained(
 pipe.to(device)
 
 # Pull pre-compiled Flux2 Transformer blocks from HF hub
+# flash-attn估计库估计更新了，导致冲突了，不使用预编译的了
 # spaces.aoti_blocks_load(pipe.transformer, "zerogpu-aoti/FLUX.2", variant="fa3")
 
-# 禁用编译，使用 SDPA 注意力
-# pipe.transformer.config._attn_implementation = "sdpa"
-# spaces.aoti_blocks_load(pipe.transformer, "zerogpu-aoti/FLUX.2", variant="sdpa")
 
 def image_to_data_uri(img):
     buffered = io.BytesIO()
@@ -128,11 +126,11 @@ def upsample_prompt_logic(prompt, image_list):
         print(f"Upsampling failed: {e}")
         return prompt
 
-def update_dimensions_from_image(image_list):
+def update_dimensions_from_image(image_list, width, height):
     """Update width/height sliders based on uploaded image aspect ratio.
     Keeps one side at 1024 and scales the other proportionally, with both sides as multiples of 8."""
     if image_list is None or len(image_list) == 0:
-        return 1024, 1024  # Default dimensions
+        return width, height  # Default dimensions
     
     # Get the first image to determine dimensions
     img = image_list[0][0]  # Gallery returns list of tuples (image, caption)
@@ -191,6 +189,9 @@ def infer(prompt, input_images=None, seed=42, randomize_seed=False, width=1024,
     
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
+
+    # Auto-update dimensions when images are uploaded
+    width, height = update_dimensions_from_image(input_images, width, height)
     
     # Prepare image list (convert None or empty gallery to None)
     image_list = None
@@ -355,13 +356,6 @@ with gr.Blocks() as demo:
             cache_mode="lazy"
         )
 
-    # Auto-update dimensions when images are uploaded
-    input_images.upload(
-        fn=update_dimensions_from_image,
-        inputs=[input_images],
-        outputs=[width, height]
-    )
-
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=infer,