diff --git a/modules/ui.py b/modules/ui.py index f019bc8..3504953 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):