From 5f5959bc33a87dcbda9bca560ad189a42ce43ad1 Mon Sep 17 00:00:00 2001 From: hiyouga Date: Mon, 20 Nov 2023 22:57:44 +0800 Subject: [PATCH] set version Former-commit-id: 6b47ad74c7b3099f9b5087c73db4aee42c451297 --- src/llmtuner/__init__.py | 2 +- src/llmtuner/webui/chatter.py | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) 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.")