From 9a064b78e67c6018b7b9b646c4199b7ea63d702e Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Wed, 5 Apr 2023 22:43:10 -0300
Subject: [PATCH] Minor simplification

---
 server.py | 17 ++---------------
 1 file changed, 2 insertions(+), 15 deletions(-)

diff --git a/server.py b/server.py
index cb93153..68cb8fd 100644
--- a/server.py
+++ b/server.py
@@ -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