From 271fb51bd63bef218cb90ac47c0dcf737a2a9947 Mon Sep 17 00:00:00 2001
From: Aptronymist <108482020+Aptronymist@users.noreply.github.com>
Date: Sat, 2 Mar 2024 13:46:15 -0500
Subject: [PATCH] Revert "1 change, 1 fix"

This reverts commit 7239de57ac04f95aa645914142eb516f0435d9ee.
---
 .vscode/settings.json  | 3 +--
 cli/install-sf.py      | 2 +-
 modules/ui_sections.py | 2 ++
 3 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/.vscode/settings.json b/.vscode/settings.json
index 1a62fb2bb..37a12179f 100644
--- a/.vscode/settings.json
+++ b/.vscode/settings.json
@@ -10,6 +10,5 @@
     "./repositories/stable-diffusion-stability-ai/ldm"
   ],
   "python.analysis.typeCheckingMode": "off",
-  "editor.formatOnSave": false,
-  "editor.tabCompletion": "on"
+  "editor.formatOnSave": false
 }
\ No newline at end of file
diff --git a/cli/install-sf.py b/cli/install-sf.py
index 894e36720..eb0d0b577 100755
--- a/cli/install-sf.py
+++ b/cli/install-sf.py
@@ -3,7 +3,7 @@ import os
 import re
 import sys
 
-torch_supported = ['211', '212','220','221']
+torch_supported = ['211', '212']
 cuda_supported = ['cu118', 'cu121']
 python_supported = ['39', '310', '311']
 repo_url = 'https://github.com/chengzeyi/stable-fast'
diff --git a/modules/ui_sections.py b/modules/ui_sections.py
index 214dce4c3..3fa9d9f4c 100644
--- a/modules/ui_sections.py
+++ b/modules/ui_sections.py
@@ -74,6 +74,8 @@ def create_batch_inputs(tab):
         with gr.Row(elem_id=f"{tab}_row_batch"):
             batch_count = gr.Slider(minimum=1, step=1, label='Batch count', value=1, elem_id=f"{tab}_batch_count")
             batch_size = gr.Slider(minimum=1, maximum=32, step=1, label='Batch size', value=1, elem_id=f"{tab}_batch_size")
+            batch_switch_btn = ToolButton(value=ui_symbols.switch, elem_id=f"{tab}_batch_switch_btn", label="Switch dims")
+            batch_switch_btn.click(lambda w, h: (h, w), inputs=[batch_count, batch_size], outputs=[batch_count, batch_size], show_progress=False)
     return batch_count, batch_size