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

Merge pull request #596 from beat4ocean/beat

Browse Source 
fix KeyError: 'lang' bug

Former-commit-id: 1968d9d1d0f64d2d332c29d3c05c87d2e79b828f
This commit is contained in:
hoshi-hiyouga 2023-08-20 16:37:40 +08:00 committed by GitHub
commit d24ece4a57
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()
if lang:
refresh_dict["lang"] = {"value": lang}
else:
refresh_dict["lang"] = {"value": user_config["lang"] if user_config["lang"] else "en"}
refresh_dict = {}
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_path"] = {"value": get_model_path(user_config["last_model"])}