Minor simplification

This commit is contained in:
oobabooga
2023-04-05 22:43:10 -03:00
parent 92ea89e59a
commit 9a064b78e6

View File

@@ -525,26 +525,13 @@ def create_interface():
if shared.args.extensions is not None: if shared.args.extensions is not None:
extensions_module.create_extensions_block() extensions_module.create_extensions_block()
def update_dict(_dict, k, v):
_dict[k] = v
return _dict
print([k for k in shared.gradio])
for k in ['do_sample', 'temperature', 'top_p', 'typical_p', 'repetition_penalty', 'encoder_repetition_penalty', 'top_k', 'min_length', 'no_repeat_ngram_size', 'num_beams', 'penalty_alpha', 'length_penalty', 'early_stopping', 'max_new_tokens', 'seed', 'stop_at_newline', 'chat_prompt_size', 'chat_generation_attempts']: for k in ['do_sample', 'temperature', 'top_p', 'typical_p', 'repetition_penalty', 'encoder_repetition_penalty', 'top_k', 'min_length', 'no_repeat_ngram_size', 'num_beams', 'penalty_alpha', 'length_penalty', 'early_stopping', 'max_new_tokens', 'seed', 'stop_at_newline', 'chat_prompt_size', 'chat_generation_attempts']:
if k not in shared.gradio: if k not in shared.gradio:
continue continue
if type(shared.gradio[k]) in [gr.Checkbox, gr.Number]: if type(shared.gradio[k]) in [gr.Checkbox, gr.Number]:
shared.gradio[k].change( shared.gradio[k].change(lambda state, value, copy=k: state.update({copy: value}), inputs=[shared.gradio['generation_state'], shared.gradio[k]], outputs=shared.gradio['generation_state'])
lambda state, value, copy=k: update_dict(state, copy, value),
inputs=[shared.gradio['generation_state'], shared.gradio[k]],
outputs=shared.gradio['generation_state'],
)
else: else:
shared.gradio[k].release( shared.gradio[k].release(lambda state, value, copy=k: state.update({copy: value}), inputs=[shared.gradio['generation_state'], shared.gradio[k]], outputs=shared.gradio['generation_state'])
lambda state, value, copy=k: update_dict(state, copy, value),
inputs=[shared.gradio['generation_state'], shared.gradio[k]],
outputs=shared.gradio['generation_state'],
)
# Authentication # Authentication
auth = None auth = None