Update app.py

app.py

@@ -125,6 +125,7 @@ tag_model.to(device, dtype=weight_dtype)
 def process(
     input_image: Image.Image,
     user_prompt: str,
+    patch_size: int,
     use_KDS: bool,
     bandwidth: float,
     num_particles: int,
@@ -179,7 +180,7 @@ def process(
                     guidance_scale=cfg_scale,  conditioning_scale=1,
                     start_point='lr', start_steps=999,ram_encoder_hidden_states=ram_encoder_hidden_states,
                     latent_tiled_size=latent_tiled_size, latent_tiled_overlap=latent_tiled_overlap, 
-                    use_KDS=use_KDS, bandwidth=bandwidth, num_particles=num_particles
+                    use_KDS=use_KDS, bandwidth=bandwidth, num_particles=num_particles, patch_size=patch_size,
                 ).images[0]
 
             if True:  # alpha<1.0:
@@ -213,6 +214,7 @@ with block:
             input_image = gr.Image(type="pil")
             num_particles = gr.Slider(label="Num of Partickes", minimum=1, maximum=16, step=1, value=10)
             bandwidth = gr.Slider(label="Bandwidth", minimum=0.1, maximum=0.8, step=0.1, value=0.1)
+            patch_size = gr.Slider(label="Patch Size", minimum=1, maximum=16, step=16, value=4)
             use_KDS = gr.Checkbox(label="Use Kernel Density Steering")  
             run_button = gr.Button("Run")
             with gr.Accordion("Options", open=True):
@@ -238,6 +240,24 @@ with block:
                         "",
                         False,
                         0.1,
+                        16,
+                        4,
+                        "clean, high-resolution, 8k, best quality, masterpiece",
+                        "dotted, noise, blur, lowres, oversmooth, longbody, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality",
+                        50,
+                        4,
+                        7.5,
+                        123,
+                        320,
+                        4,
+                        1,
+                    ],
+                    [
+                        "preset/datasets/test_datasets/179.png", 
+                        "",
+                        True,
+                        0.1,
+                        16,
                         4,
                         "clean, high-resolution, 8k, best quality, masterpiece",
                         "dotted, noise, blur, lowres, oversmooth, longbody, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality",
@@ -255,6 +275,24 @@ with block:
                         True,
                         0.1,
                         4,
+                        4,
+                        "clean, high-resolution, 8k, best quality, masterpiece",
+                        "dotted, noise, blur, lowres, oversmooth, longbody, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality",
+                        50,
+                        4,
+                        7.5,
+                        123,
+                        320,
+                        4,
+                        1,
+                    ],
+                    [
+                        "preset/datasets/test_datasets/179.png", 
+                        "",
+                        True,
+                        0.1,
+                        4,
+                        1,
                         "clean, high-resolution, 8k, best quality, masterpiece",
                         "dotted, noise, blur, lowres, oversmooth, longbody, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality",
                         50,
@@ -270,6 +308,7 @@ with block:
                         "",
                         True,
                         0.1,
+                        32,
                         16,
                         "clean, high-resolution, 8k, best quality, masterpiece",
                         "dotted, noise, blur, lowres, oversmooth, longbody, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality",
@@ -286,6 +325,7 @@ with block:
                     input_image,
                     user_prompt,
                     use_KDS,
+                    patch_size,
                     bandwidth,
                     num_particles,
                     positive_prompt,
@@ -306,6 +346,7 @@ with block:
         input_image,
         user_prompt,
         use_KDS,
+        patch_size,
         bandwidth,
         num_particles,
         positive_prompt,