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

fix #595

Browse Source 
Former-commit-id: 5c052836a0a19afd556418f1f6da8db8fc3cd37d
This commit is contained in:
hiyouga 2023-08-20 16:40:00 +08:00
parent d24ece4a57
commit fa15416024
2 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/llmtuner/webui/interface.py
View File

@ -59,7 +59,7 @@ def create_web_demo() -> gr.Blocks:
chat_model = WebChatModel(lazy_init=False) chat_model = WebChatModel(lazy_init=False)
with gr.Blocks(title="Web Demo", css=CSS) as demo: with gr.Blocks(title="Web Demo", css=CSS) as demo:
lang = gr.Dropdown(choices=["en", "zh"]) lang = gr.Dropdown(choices=["en", "zh"], value="en")
_, _, _, chat_elems = create_chat_box(chat_model, visible=True) _, _, _, chat_elems = create_chat_box(chat_model, visible=True)

13
src/llmtuner/webui/manager.py
View File

@ -19,14 +19,15 @@ class Manager:
} }
user_config = load_config() user_config = load_config()
refresh_dict = {} if not lang:
if user_config.get("lang", None):
if "lang" in user_config: lang = user_config["lang"]
refresh_dict["lang"] = {"value": user_config["lang"]}
else: else:
refresh_dict["lang"] = {"value": "en"} lang = "en"
if user_config.get("last_model"): refresh_dict["lang"] = {"value": lang}
if user_config.get("last_model", None):
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"])}