1 change, 1 fix

Fixes install-sf.py not supporting torch >=2.2.0
Removed batch switch button. Useless. 0.0001% of people use it. 😀

.vscode/settings.json

@@ -10,5 +10,6 @@
     "./repositories/stable-diffusion-stability-ai/ldm"
   ],
   "python.analysis.typeCheckingMode": "off",
-  "editor.formatOnSave": false
+  "editor.formatOnSave": false,
+  "editor.tabCompletion": "on"
 }

cli/install-sf.py

@@ -3,7 +3,7 @@ import os
 import re
 import sys
 
-torch_supported = ['211', '212']
+torch_supported = ['211', '212','220','221']
 cuda_supported = ['cu118', 'cu121']
 python_supported = ['39', '310', '311']
 repo_url = 'https://github.com/chengzeyi/stable-fast'

modules/ui_sections.py

@@ -74,8 +74,6 @@ 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