From 7b45de6b9f5a0159335614e78a106b595d726fc9 Mon Sep 17 00:00:00 2001
From: beat4ocean <beat4ocean@gmail.com>
Date: Sun, 20 Aug 2023 15:32:36 +0800
Subject: [PATCH] fix KeyError: 'lang' bug

---
 src/llmtuner/webui/manager.py | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/src/llmtuner/webui/manager.py b/src/llmtuner/webui/manager.py
index 2d5a0a39..63a4fa5b 100644
--- a/src/llmtuner/webui/manager.py
+++ b/src/llmtuner/webui/manager.py
@@ -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"])}