tykiww commited on
Commit
8d7ed18
·
verified ·
1 Parent(s): cbbdafb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -81,9 +81,6 @@ def main():
81
  value=peftparams['seed'])
82
  rslora = gr.Textbox(label="Use R-S LoRA",
83
  value=peftparams['rslora'])
84
- pefts = [r, alpha, dropout, bias, seed, rslora]
85
-
86
- gr.Markdown("List of items")
87
  gr.Markdown("### Supervised Fine-Tuning Parameters")
88
  gr.Markdown("List of items")
89
 
@@ -95,7 +92,13 @@ def main():
95
  output = gr.Textbox(label="Output")
96
  # Execute button
97
  tune_btn.click(fn=greet,
98
- inputs=[model_name, inject_prompt, dataset_predefined, pefts],
 
 
 
 
 
 
99
  outputs=output)
100
  # Launch baby
101
  demo.launch()
 
81
  value=peftparams['seed'])
82
  rslora = gr.Textbox(label="Use R-S LoRA",
83
  value=peftparams['rslora'])
 
 
 
84
  gr.Markdown("### Supervised Fine-Tuning Parameters")
85
  gr.Markdown("List of items")
86
 
 
92
  output = gr.Textbox(label="Output")
93
  # Execute button
94
  tune_btn.click(fn=greet,
95
+ inputs=[model_name,
96
+ inject_prompt,
97
+ dataset_predefined,
98
+ # peft
99
+ r, alpha, dropout, bias, seed, rslora
100
+ # sft
101
+ ],
102
  outputs=output)
103
  # Launch baby
104
  demo.launch()