llama board: add unsloth

Former-commit-id: 9477e6f28808ae9deadada1f6cf679a29542c271
2023-12-23 00:35:53 +08:00
parent 6faf9c35a9
commit 0fdd6074c3
5 changed files with 13 additions and 26 deletions
--- a/src/llmtuner/webui/locales.py
+++ b/src/llmtuner/webui/locales.py
@@ -85,20 +85,12 @@ LOCALES = {
            "label": "RoPE 插值方法"
        }
    },
-    "flash_attn": {
+    "booster": {
        "en": {
-            "label": "Use FlashAttention-2"
+            "label": "Booster"
        },
        "zh": {
-            "label": "使用 FlashAttention-2"
-        }
-    },
-    "shift_attn": {
-        "en": {
-            "label": "Use shift short attention (S^2-Attn)"
-        },
-        "zh": {
-            "label": "使用 shift short attention (S^2-Attn)"
+            "label": "加速方式"
        }
    },
    "training_stage": {