diff --git a/modules/ui.py b/modules/ui.py
index b1a8c776..5eb12b02 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -228,7 +228,7 @@ def create_ui(txt2img, img2img, run_extras, run_pnginfo):
             with gr.Column(variant='panel'):
                 with gr.Group():
                     txt2img_preview = gr.Image(elem_id='txt2img_preview', visible=False)
-                    txt2img_gallery = gr.Gallery(label='Output', elem_id='txt2img_gallery')
+                    txt2img_gallery = gr.Gallery(label='Output', elem_id='txt2img_gallery').style(grid=4)
 
 
                 with gr.Group():
@@ -363,7 +363,7 @@ def create_ui(txt2img, img2img, run_extras, run_pnginfo):
             with gr.Column(variant='panel'):
                 with gr.Group():
                     img2img_preview = gr.Image(elem_id='img2img_preview', visible=False)
-                    img2img_gallery = gr.Gallery(label='Output', elem_id='img2img_gallery')
+                    img2img_gallery = gr.Gallery(label='Output', elem_id='img2img_gallery').style(grid=4)
 
                 with gr.Group():
                     with gr.Row():
diff --git a/webui.bat b/webui.bat
index 36e98228..1d49fe4a 100644
--- a/webui.bat
+++ b/webui.bat
@@ -2,7 +2,7 @@
 
 set PYTHON=python
 set GIT=git
-set COMMANDLINE_ARGS=
+set COMMANDLINE_ARGS=%*
 set VENV_DIR=venv
 
 if exist webui.settings.bat (