fix webui export model

Former-commit-id: c34469c05e681239db23e2e666b5ac6a4e38aba9
2023-08-04 14:20:27 +08:00
parent e2a4e926b9
commit aaa85190eb
2 changed files with 9 additions and 1 deletions
--- a/src/llmtuner/webui/components/export.py
+++ b/src/llmtuner/webui/components/export.py
@@ -22,6 +22,7 @@ def create_export_tab(top_elems: Dict[str, "Component"]) -> Dict[str, "Component
            top_elems["model_name"],
            top_elems["checkpoints"],
            top_elems["finetuning_type"],
+            top_elems["template"],
            max_shard_size,
            save_dir
        ],