From 9697c3e970a9ea9ca58bdd4eef9b18671e41a8e0 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: 35c2da3eba064e16b21c20a4cde3355173d5d9fd --- 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.")