update webui #1086

2023-10-09 14:50:14 +08:00
parent a683c5b797
commit b8dbec086e
10 changed files with 105 additions and 56 deletions
--- a/src/llmtuner/webui/components/eval.py
+++ b/src/llmtuner/webui/components/eval.py
@@ -57,6 +57,9 @@ def create_eval_tab(top_elems: Dict[str, "Component"], runner: "Runner") -> Dict
        top_elems["quantization_bit"],
        top_elems["template"],
        top_elems["system_prompt"],
+        top_elems["flash_attn"],
+        top_elems["shift_attn"],
+        top_elems["rope_scaling"],
        dataset_dir,
        dataset,
        cutoff_len,