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

support save args in webui #2807 #3046

Browse Source 
some ideas are borrowed from @marko1616
This commit is contained in:
hiyouga
2024-03-30 23:09:12 +08:00
parent 257f643a74
commit 7a086ed333
9 changed files with 219 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,5 +1,4 @@
import gradio as gr
from transformers.utils.versions import require_version
from .common import save_config
from .components import (
@@ -14,9 +13,6 @@ from .css import CSS
from .engine import Engine
require_version("gradio>4.0.0,<=4.21.0", "To fix: pip install gradio==4.21.0")
def create_ui(demo_mode: bool = False) -> gr.Blocks:
engine = Engine(demo_mode=demo_mode, pure_chat=False)
@@ -29,21 +25,21 @@ def create_ui(demo_mode: bool = False) -> gr.Blocks:
)
gr.DuplicateButton(value="Duplicate Space for private use", elem_classes="duplicate-button")
engine.manager.add_elem_dict("top", create_top())
engine.manager.add_elems("top", create_top())
lang: "gr.Dropdown" = engine.manager.get_elem_by_id("top.lang")
with gr.Tab("Train"):
engine.manager.add_elem_dict("train", create_train_tab(engine))
engine.manager.add_elems("train", create_train_tab(engine))
with gr.Tab("Evaluate & Predict"):
engine.manager.add_elem_dict("eval", create_eval_tab(engine))
engine.manager.add_elems("eval", create_eval_tab(engine))
with gr.Tab("Chat"):
engine.manager.add_elem_dict("infer", create_infer_tab(engine))
engine.manager.add_elems("infer", create_infer_tab(engine))
if not demo_mode:
with gr.Tab("Export"):
engine.manager.add_elem_dict("export", create_export_tab(engine))
engine.manager.add_elems("export", create_export_tab(engine))
demo.load(engine.resume, outputs=engine.manager.get_elem_list(), concurrency_limit=None)
lang.change(engine.change_lang, [lang], engine.manager.get_elem_list(), queue=False)
@@ -57,10 +53,10 @@ def create_web_demo() -> gr.Blocks:
with gr.Blocks(title="Web Demo", css=CSS) as demo:
lang = gr.Dropdown(choices=["en", "zh"])
engine.manager.add_elem_dict("top", dict(lang=lang))
engine.manager.add_elems("top", dict(lang=lang))
chat_box, _, _, chat_elems = create_chat_box(engine, visible=True)
engine.manager.add_elem_dict("infer", dict(chat_box=chat_box, **chat_elems))
engine.manager.add_elems("infer", dict(chat_box=chat_box, **chat_elems))
demo.load(engine.resume, outputs=engine.manager.get_elem_list(), concurrency_limit=None)
lang.change(engine.change_lang, [lang], engine.manager.get_elem_list(), queue=False)