From 7e2f7bd96cde02cd0ebd8c5ac2354bf9ac4836ff Mon Sep 17 00:00:00 2001
From: hiyouga <hiyouga@buaa.edu.cn>
Date: Tue, 20 Feb 2024 11:12:43 +0800
Subject: [PATCH] release v0.5.2

Former-commit-id: 5bbec1c1f26c29e5dbbab7eca64179c9c67a2dc2
---
 src/llmtuner/__init__.py               |  2 +-
 src/llmtuner/webui/components/train.py | 13 ++++++++++++-
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/src/llmtuner/__init__.py b/src/llmtuner/__init__.py
index 96cde64c..115fdb27 100644
--- a/src/llmtuner/__init__.py
+++ b/src/llmtuner/__init__.py
@@ -7,5 +7,5 @@ from .train import export_model, run_exp
 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
index 4f942cf8..798abbfb 100644
--- 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,