Skip to content

Commit 1c55186

Browse files
committed
support custom quantized dataset (#1893)
(cherry picked from commit a0ff73c)
1 parent 643b9b7 commit 1c55186

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

swift/ui/llm_export/export.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,4 +91,9 @@ def do_build_ui(cls, base_tab: Type['BaseUI']):
9191
gr.Textbox(elem_id='quant_seqlen', scale=20)
9292
with gr.Row():
9393
gr.Textbox(elem_id='quant_output_dir', scale=20)
94-
gr.Dropdown(elem_id='dataset', multiselect=True, choices=list(DATASET_MAPPING.keys()), scale=20)
94+
gr.Dropdown(
95+
elem_id='dataset',
96+
multiselect=True,
97+
allow_custom_value=True,
98+
choices=list(DATASET_MAPPING.keys()),
99+
scale=20)

0 commit comments

Comments
 (0)