mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Minor change to ui.py

Browse source
This commit is contained in:
oobabooga 2023-07-07 09:09:14 -07:00
parent de994331a4
commit c21b73ff37
1 changed files with 1 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

5
modules/ui.py
View file

@ -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):