tykiww commited on
Commit
611772e
·
verified ·
1 Parent(s): dc813e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -1
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")