modify code structure

Former-commit-id: 6369f9b1751e6f9bb709ba76a85f69cbe0823e5d
2023-08-02 23:17:36 +08:00
parent 8bd1da7144
commit 28a51b622b
25 changed files with 188 additions and 145 deletions
--- a/src/llmtuner/webui/components/export.py
+++ b/src/llmtuner/webui/components/export.py
@@ -1,7 +1,7 @@
 from typing import TYPE_CHECKING, Dict
 import gradio as gr

-from llmtuner.webui.utils import export_model
+from llmtuner.webui.utils import save_model

 if TYPE_CHECKING:
    from gradio.components import Component
@@ -16,7 +16,7 @@ def create_export_tab(top_elems: Dict[str, "Component"]) -> Dict[str, "Component
    info_box = gr.Textbox(show_label=False, interactive=False)

    export_btn.click(
-        export_model,
+        save_model,
        [
            top_elems["lang"],
            top_elems["model_name"],