adjust chopping size and chopping batchsize

app.py

@@ -54,6 +54,12 @@ def get_configs(num_steps=1, chopping_size=128, seed=12345):
     configs.bs = 1
     configs.seed = 12345
     configs.basesr.chopping.pch_size = chopping_size
+    if chopping_size == 128:
+        configs.basesr.chopping.extra_bs = 8
+    elif chopping_size == 256:
+        configs.basesr.chopping.extra_bs = 4
+    else:
+        configs.basesr.chopping.extra_bs = 1
 
     return configs
 
@@ -114,7 +120,7 @@ demo = gr.Interface(
             label="Number of steps",
             ),
         gr.Dropdown(
-            choices=[128, 256],
+            choices=[128, 256, 512],
             value=128,
             label="Chopping size",
             ),