From 4ac36c14ae8b1f6e4fa5da55f35b20b4f6480fce Mon Sep 17 00:00:00 2001 From: hiyouga Date: Sat, 20 Jan 2024 20:21:39 +0800 Subject: [PATCH] release v0.5.0 Former-commit-id: 71cfdc2658e346e3fd05b3a8a678c811b3ba1bd5 --- src/llmtuner/__init__.py | 12 ++++++------ src/llmtuner/data/__init__.py | 2 +- src/llmtuner/hparams/__init__.py | 4 ++-- src/llmtuner/webui/components/__init__.py | 8 ++++---- 4 files changed, 13 insertions(+), 13 deletions(-) diff --git a/src/llmtuner/__init__.py b/src/llmtuner/__init__.py index 512b3fbc..670adbc0 100644 --- a/src/llmtuner/__init__.py +++ b/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"] diff --git a/src/llmtuner/data/__init__.py b/src/llmtuner/data/__init__.py index c941faca..aee03970 100644 --- a/src/llmtuner/data/__init__.py +++ b/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"] diff --git a/src/llmtuner/hparams/__init__.py b/src/llmtuner/hparams/__init__.py index ef062ac6..d1ee98dd 100644 --- a/src/llmtuner/hparams/__init__.py +++ b/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", ] diff --git a/src/llmtuner/webui/components/__init__.py b/src/llmtuner/webui/components/__init__.py index 1352af8e..5c1e21b8 100644 --- a/src/llmtuner/webui/components/__init__.py +++ b/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", ]