|
|
@@ -525,26 +525,13 @@ def create_interface():
|
|
|
if shared.args.extensions is not None:
|
|
|
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']:
|
|
|
if k not in shared.gradio:
|
|
|
continue
|
|
|
if type(shared.gradio[k]) in [gr.Checkbox, gr.Number]:
|
|
|
- shared.gradio[k].change(
|
|
|
- lambda state, value, copy=k: update_dict(state, copy, value),
|
|
|
- inputs=[shared.gradio['generation_state'], shared.gradio[k]],
|
|
|
- outputs=shared.gradio['generation_state'],
|
|
|
- )
|
|
|
+ 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'])
|
|
|
else:
|
|
|
- shared.gradio[k].release(
|
|
|
- lambda state, value, copy=k: update_dict(state, copy, value),
|
|
|
- inputs=[shared.gradio['generation_state'], shared.gradio[k]],
|
|
|
- outputs=shared.gradio['generation_state'],
|
|
|
- )
|
|
|
+ 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'])
|
|
|
|
|
|
# Authentication
|
|
|
auth = None
|
oobabooga