From c21b73ff37d9d2613509dd1238aca15fd327510e Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Fri, 7 Jul 2023 09:09:14 -0700 Subject: [PATCH] Minor change to ui.py --- modules/ui.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/modules/ui.py b/modules/ui.py index f019bc87..35049533 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -67,7 +67,6 @@ def list_model_elements(): 'alpha_value' ] - for i in range(torch.cuda.device_count()): elements.append(f'gpu_memory_{i}') @@ -76,7 +75,6 @@ def list_model_elements(): def list_interface_input_elements(): elements = [ - 'preset_menu', 'max_new_tokens', 'seed', 'temperature', @@ -158,8 +156,7 @@ def apply_interface_values(state, use_persistent=False): if len(state) == 0: return [gr.update() for k in elements] # Dummy, do nothing else: - ans = [state[k] if k in state else gr.update() for k in elements] - return ans + return [state[k] if k in state else gr.update() for k in elements] class ToolButton(gr.Button, gr.components.FormComponent):