p04798526
/
LLaMA-Factory-310P3
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #596 from beat4ocean/beat 

fix KeyError: 'lang' bug
This commit is contained in:
hoshi-hiyouga 2023-08-20 16:37:40 +08:00 committed by GitHub
parent 0676497104 7b45de6b9f
commit 1968d9d1d0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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"])}