423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-22 22:02:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

set version

Browse Source 
Former-commit-id: 35c2da3eba064e16b21c20a4cde3355173d5d9fd
This commit is contained in:
hiyouga 2023-11-20 22:57:44 +08:00
parent 4966bd7911
commit 9697c3e970
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/llmtuner/__init__.py
View File

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

3
src/llmtuner/webui/chatter.py
View File

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