From 7239de57ac04f95aa645914142eb516f0435d9ee Mon Sep 17 00:00:00 2001 From: Aptronymist <108482020+Aptronymist@users.noreply.github.com> Date: Sat, 2 Mar 2024 13:09:46 -0500 Subject: [PATCH] 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. :grinning: --- .vscode/settings.json | 3 ++- cli/install-sf.py | 2 +- modules/ui_sections.py | 2 -- 3 files changed, 3 insertions(+), 4 deletions(-) diff --git a/.vscode/settings.json b/.vscode/settings.json index 37a12179f..1a62fb2bb 100644 --- a/.vscode/settings.json +++ b/.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" } \ No newline at end of file diff --git a/cli/install-sf.py b/cli/install-sf.py index eb0d0b577..894e36720 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'] +torch_supported = ['211', '212','220','221'] 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 3fa9d9f4c..214dce4c3 100644 --- a/modules/ui_sections.py +++ b/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