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

fix KeyError: 'lang' bug

Browse Source 
Former-commit-id: 4d4d9172b1f362cb4876315f1f5739e417055065
This commit is contained in:
beat4ocean 2023-08-20 15:32:36 +08:00
parent 5549f35939
commit 713fa28804
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/llmtuner/webui/manager.py
View File

@ -19,12 +19,14 @@ class Manager:
} }
user_config = load_config() user_config = load_config()
if lang: refresh_dict = {}
refresh_dict["lang"] = {"value": lang}
else:
refresh_dict["lang"] = {"value": user_config["lang"] if user_config["lang"] else "en"}
if user_config["last_model"]: if "lang" in user_config:
refresh_dict["lang"] = {"value": user_config["lang"]}
else:
refresh_dict["lang"] = {"value": "en"}
if user_config.get("last_model"):
refresh_dict["model_name"] = {"value": user_config["last_model"]} refresh_dict["model_name"] = {"value": user_config["last_model"]}
refresh_dict["model_path"] = {"value": get_model_path(user_config["last_model"])} refresh_dict["model_path"] = {"value": get_model_path(user_config["last_model"])}