fix webui

Former-commit-id: 2c8b7414be9b43e20cc1d0575cc4dc1c7545fd86
2023-08-12 23:52:07 +08:00
parent a22982f2fa
commit 6c7225a5d4
3 changed files with 3 additions and 5 deletions
--- a/src/llmtuner/webui/components/top.py
+++ b/src/llmtuner/webui/components/top.py
@@ -26,7 +26,7 @@ def create_top() -> Dict[str, "Component"]:

    with gr.Accordion(label="Advanced config", open=False) as advanced_tab:
        with gr.Row():
-            quantization_bit = gr.Dropdown([8, 4], scale=1)
+            quantization_bit = gr.Dropdown(["", "8", "4"], scale=1)
            template = gr.Dropdown(value="default", choices=list(templates.keys()), scale=1)
            source_prefix = gr.Textbox(scale=2)