diff --git a/src/llmtuner/__init__.py b/src/llmtuner/__init__.py index 712be7c2..27b96d01 100644 --- a/src/llmtuner/__init__.py +++ b/src/llmtuner/__init__.py @@ -7,4 +7,4 @@ from llmtuner.train import export_model, run_exp from llmtuner.webui import create_ui, create_web_demo -__version__ = "0.3.1" +__version__ = "0.3.2" diff --git a/src/llmtuner/webui/chatter.py b/src/llmtuner/webui/chatter.py index 10c06839..d591a2e3 100644 --- a/src/llmtuner/webui/chatter.py +++ b/src/llmtuner/webui/chatter.py @@ -34,7 +34,10 @@ class WebChatModel(ChatModel): try: with open("demo_config.json", "r", encoding="utf-8") as f: args = json.load(f) + assert args.get("model_name_or_path", None) and args.get("template", None) super().__init__(args) + except AssertionError: + print("Please provided model name and template in `demo_config.json`.") except: print("Cannot find `demo_config.json` at current directory.")