ysharma HF Staff commited on
Commit
d83288b
·
1 Parent(s): 12bf04e

update layout

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -72,11 +72,11 @@ with gr.Blocks() as demo:
72
  with gr.Row():
73
  out_image = gr.Image(label="Image generated by LORA model")
74
  with gr.Column():
75
- out_file = gr.File(label="Lora trained model weights", )
76
  with gr.Accordion("Advance settings for Training and Inference", open=False):
77
  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.")
78
  in_steps = gr.Number(label="Enter the number of training steps", value = 4000)
79
  in_alpha = gr.Slider(0.1,1.0, step=0.01, label="Set Alpha level", value=0.5)
 
80
 
81
  b1.click(fn = accelerate_train_lora, inputs=in_steps, outputs=out_file)
82
  b2.click(fn = monkeypatching, inputs=[in_alpha, in_prompt], outputs=out_image)
 
72
  with gr.Row():
73
  out_image = gr.Image(label="Image generated by LORA model")
74
  with gr.Column():
 
75
  with gr.Accordion("Advance settings for Training and Inference", open=False):
76
  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.")
77
  in_steps = gr.Number(label="Enter the number of training steps", value = 4000)
78
  in_alpha = gr.Slider(0.1,1.0, step=0.01, label="Set Alpha level", value=0.5)
79
+ out_file = gr.File(label="Lora trained model weights", )
80
 
81
  b1.click(fn = accelerate_train_lora, inputs=in_steps, outputs=out_file)
82
  b2.click(fn = monkeypatching, inputs=[in_alpha, in_prompt], outputs=out_image)