diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index 73117c3b..349e4373 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -40,6 +40,7 @@ jobs: - name: Install dependencies run: | python -m pip install --upgrade pip + python -m pip install git+https://github.com/huggingface/transformers.git python -m pip install ".[torch,dev]" - name: Check quality diff --git a/src/llamafactory/webui/components/top.py b/src/llamafactory/webui/components/top.py index 55096601..2cec4f75 100644 --- a/src/llamafactory/webui/components/top.py +++ b/src/llamafactory/webui/components/top.py @@ -43,11 +43,11 @@ def create_top() -> Dict[str, "Component"]: with gr.Accordion(open=False) as advanced_tab: with gr.Row(): - quantization_bit = gr.Dropdown(choices=["none", "8", "4"], value="none", allow_custom_value=True, scale=1) - quantization_method = gr.Dropdown(choices=["bitsandbytes", "hqq", "eetq"], value="bitsandbytes", scale=1) - template = gr.Dropdown(choices=list(TEMPLATES.keys()), value="default", scale=1) - rope_scaling = gr.Radio(choices=["none", "linear", "dynamic"], value="none", scale=2) - booster = gr.Radio(choices=["auto", "flashattn2", "unsloth", "liger_kernel"], value="auto", scale=3) + quantization_bit = gr.Dropdown(choices=["none", "8", "4"], value="none", allow_custom_value=True, scale=2) + quantization_method = gr.Dropdown(choices=["bitsandbytes", "hqq", "eetq"], value="bitsandbytes", scale=2) + template = gr.Dropdown(choices=list(TEMPLATES.keys()), value="default", scale=2) + rope_scaling = gr.Radio(choices=["none", "linear", "dynamic"], value="none", scale=3) + booster = gr.Radio(choices=["auto", "flashattn2", "unsloth", "liger_kernel"], value="auto", scale=5) model_name.change(get_model_info, [model_name], [model_path, template], queue=False).then( list_checkpoints, [model_name, finetuning_type], [checkpoint_path], queue=False diff --git a/src/llamafactory/webui/locales.py b/src/llamafactory/webui/locales.py index f742c246..852c6f23 100644 --- a/src/llamafactory/webui/locales.py +++ b/src/llamafactory/webui/locales.py @@ -148,7 +148,7 @@ LOCALES = { }, "zh": { "label": "提示模板", - "info": "构建提示词时使用的模板", + "info": "构建提示词时使用的模板。", }, "ko": { "label": "프롬프트 템플릿",