Update app.py
Browse files
app.py
CHANGED
@@ -63,9 +63,20 @@ def main():
|
|
63 |
dataset_choice,
|
64 |
dataset_upload)
|
65 |
# Hyperparameters (allow selection, but hide in accordion.)
|
66 |
-
# gr.Accordion('label')
|
67 |
with gr.Accordion("Advanced Tuning", open=False):
|
|
|
|
|
|
|
|
|
68 |
gr.Markdown("### PEFT Parameters")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
69 |
gr.Markdown("List of items")
|
70 |
gr.Markdown("### Supervised Fine-Tuning Parameters")
|
71 |
gr.Markdown("List of items")
|
|
|
63 |
dataset_choice,
|
64 |
dataset_upload)
|
65 |
# Hyperparameters (allow selection, but hide in accordion.)
|
|
|
66 |
with gr.Accordion("Advanced Tuning", open=False):
|
67 |
+
# config
|
68 |
+
peftparams = conf['model']['peft']
|
69 |
+
sftparams = conf['model']['sft']
|
70 |
+
# accordion container content
|
71 |
gr.Markdown("### PEFT Parameters")
|
72 |
+
r = gr.Textbox(label="r", value=peftparams['r'])
|
73 |
+
alpha = gr.Textbox(label="LoRA alpha", value=peftparams['alpha'])
|
74 |
+
dropout = gr.Textbox(label="LoRA dropout", value=peftparams['dropout'])
|
75 |
+
bias = gr.Textbox(label="Bias", value=peftparams['bias'])
|
76 |
+
seed = gr.Textbox(label="Random State", value=peftparams['seed'])
|
77 |
+
rslora = gr.Textbox(label="Use R-S LoRA", value=peftparams['rslora'])
|
78 |
+
pefts = [r, alpha, dropout, bias, seed, rslora]
|
79 |
+
|
80 |
gr.Markdown("List of items")
|
81 |
gr.Markdown("### Supervised Fine-Tuning Parameters")
|
82 |
gr.Markdown("List of items")
|