set version
This commit is contained in:
1 parent
9ea9380145
commit
35c2da3eba
2 files changed
+4
-1
No files matched your search
@@ -7,4 +7,4 @@ from llmtuner.train import export_model, run_exp
|
|||||||
from llmtuner.webui import create_ui, create_web_demo
|
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:
|
try:
|
||||||
with open("demo_config.json", "r", encoding="utf-8") as f:
|
with open("demo_config.json", "r", encoding="utf-8") as f:
|
||||||
args = json.load(f)
|
args = json.load(f)
|
||||||
|
assert args.get("model_name_or_path", None) and args.get("template", None)
|
||||||
super().__init__(args)
|
super().__init__(args)
|
||||||
|
except AssertionError:
|
||||||
|
print("Please provided model name and template in `demo_config.json`.")
|
||||||
except:
|
except:
|
||||||
print("Cannot find `demo_config.json` at current directory.")
|
print("Cannot find `demo_config.json` at current directory.")
|
||||||
|
|
||||||
|
|||||||
Reference in new issue
Block a user