set version
This commit is contained in:
parent
9ea9380145
commit
35c2da3eba
|
@ -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"
|
||||
|
|
|
@ -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.")
|
||||
|
||||
|
|
Loading…
Reference in New Issue