release v0.5.0

src/llmtuner/__init__.py

@@ -1,11 +1,11 @@
 # Level: api, webui > chat, eval, train > data, model > extras, hparams
 
-from llmtuner.api import create_app
-from llmtuner.chat import ChatModel
-from llmtuner.eval import Evaluator
-from llmtuner.train import export_model, run_exp
-from llmtuner.webui import create_ui, create_web_demo
+from .api import create_app
+from .chat import ChatModel
+from .eval import Evaluator
+from .train import export_model, run_exp
+from .webui import create_ui, create_web_demo
 
 
-__version__ = "0.4.0"
+__version__ = "0.5.0"
 __all__ = ["create_app", "ChatModel", "Evaluator", "export_model", "run_exp", "create_ui", "create_web_demo"]

src/llmtuner/data/__init__.py

@@ -3,4 +3,4 @@ from .template import get_template_and_fix_tokenizer, templates
 from .utils import Role, split_dataset
 
 
-__all__ = ["get_dataset", "get_template_and_fix_tokenizer", "templates", "split_dataset", "Role"]
+__all__ = ["get_dataset", "get_template_and_fix_tokenizer", "templates", "Role", "split_dataset"]

src/llmtuner/hparams/__init__.py

@@ -12,7 +12,7 @@ __all__ = [
     "FinetuningArguments",
     "GeneratingArguments",
     "ModelArguments",
-    "get_train_args",
-    "get_infer_args",
     "get_eval_args",
+    "get_infer_args",
+    "get_train_args",
 ]

src/llmtuner/webui/components/__init__.py

@@ -7,10 +7,10 @@ from .train import create_train_tab
 
 
 __all__ = [
+    "create_chat_box",
+    "create_eval_tab",
+    "create_export_tab",
+    "create_infer_tab",
     "create_top",
     "create_train_tab",
-    "create_eval_tab",
-    "create_infer_tab",
-    "create_export_tab",
-    "create_chat_box",
 ]