p12475398
/
LLaMA-Factory-Mirror
forked from p04798526/LLaMA-Factory-Mirror
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix KeyError: 'lang' bug

This commit is contained in:
beat4ocean 2023-08-20 15:32:36 +08:00
parent 0676497104
commit 7b45de6b9f
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"])}