ysharma HF staff commited on
Commit
1e291a5
·
1 Parent(s): fa1e420

update accordion

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -64,9 +64,10 @@ with gr.Blocks() as demo:
64
 
65
  with gr.Row():
66
  in_images = gr.File(label="Upload images to fine-tune for LORA", file_count="multiple")
67
- #in_prompt = gr.Textbox(label="Enter a ")
68
- in_steps = gr.Number(label="Enter number of steps", value = 4000)
69
- in_alpha = gr.Slider(0.1,1.0, step=0.01, label="Set Alpha level - higher value has more chances to overfit", value=0.5)
 
70
  with gr.Row():
71
  b1 = gr.Button(value="Train LORA model")
72
  b2 = gr.Button(value="Inference using LORA model")
 
64
 
65
  with gr.Row():
66
  in_images = gr.File(label="Upload images to fine-tune for LORA", file_count="multiple")
67
+ with gr.Accordion("Advance settings"):
68
+ gr.Markdown("Advance settings for a number of Training Steps and Alpha. Set alpha to 1.0 to fully add LORA. If the LORA seems to have too much effect (i.e., overfitting), set alpha to a lower value. If the LORA seems to have too little effect, set the alpha higher. You can tune these two values to your needs.")
69
+ in_steps = gr.Number(label="Enter the number of training steps", value = 4000)
70
+ in_alpha = gr.Slider(0.1,1.0, step=0.01, label="Set Alpha level", value=0.5)
71
  with gr.Row():
72
  b1 = gr.Button(value="Train LORA model")
73
  b2 = gr.Button(value="Inference using LORA model")