set version

This commit is contained in:
hiyouga 2023-11-20 22:57:44 +08:00
parent 9ea9380145
commit 35c2da3eba
2 changed files with 4 additions and 1 deletions

View File

@ -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"

View File

@ -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.")