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

fix KeyError: 'lang' bug

Browse Source 
Former-commit-id: 7b45de6b9f5a0159335614e78a106b595d726fc9
This commit is contained in:
beat4ocean 2023-08-20 15:32:36 +08:00
parent 570ccc3618
commit c81d1d40fa
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"])}