release v0.5.2

hiyouga · Feb 20, 2024 · 5bbec1c · 5bbec1c
1 parent ba998c6
commit 5bbec1c
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/src/llmtuner/__init__.py b/src/llmtuner/__init__.py
@@ -7,5 +7,5 @@
 from .webui import create_ui, create_web_demo
 
 
-__version__ = "0.5.1"
+__version__ = "0.5.2"
 __all__ = ["create_app", "ChatModel", "Evaluator", "export_model", "run_exp", "create_ui", "create_web_demo"]
diff --git a/src/llmtuner/webui/components/train.py b/src/llmtuner/webui/components/train.py
@@ -82,7 +82,18 @@ def create_train_tab(engine: "Engine") -> Dict[str, "Component"]:
             upcast_layernorm = gr.Checkbox()
             use_llama_pro = gr.Checkbox()
 
-    input_elems.update({logging_steps, save_steps, warmup_steps, neftune_alpha, resize_vocab, sft_packing, upcast_layernorm, use_llama_pro})
+    input_elems.update(
+        {
+            logging_steps,
+            save_steps,
+            warmup_steps,
+            neftune_alpha,
+            resize_vocab,
+            sft_packing,
+            upcast_layernorm,
+            use_llama_pro,
+        }
+    )
     elem_dict.update(
         dict(
             extra_tab=extra_tab,