max area

app.py

@@ -77,6 +77,7 @@ def infer(
     randomize_seed,
     width,
     height,
+    max_area,
     guidance_scale,      # 目前没传入 pipeline，如需要可在下面调用里加上
     num_inference_steps,
     progress=gr.Progress(track_tqdm=True),
@@ -107,7 +108,7 @@ def infer(
             height=height,
             width=width,
             num_inference_steps=int(num_inference_steps),
-            max_area=MAX_IMAGE_SIZE ** 2,
+            max_area=max_area ** 2,
             generator=generator,
             # 如需 guidance_scale，确保 pipeline 支持这个参数后再打开：
             # guidance_scale=float(guidance_scale),
@@ -180,6 +181,13 @@ with gr.Blocks(css=css) as demo:
                     step=32,
                     value=512,
                 )
+                max_area_comp = gr.Slider(
+                    label="Max Area",
+                    minimum=512,
+                    maximum=1024,
+                    step=512,
+                    value=512,
+                )
 
             with gr.Row():
                 guidance_scale_comp = gr.Slider(
@@ -217,6 +225,7 @@ with gr.Blocks(css=css) as demo:
             randomize_seed_comp,
             width_comp,
             height_comp,
+            max_area_comp,
             guidance_scale_comp,
             num_inference_steps_comp,
         ],