Spaces:
Configuration error
Configuration error
update accordion
Browse files
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 |
-
|
68 |
-
|
69 |
-
|
|
|
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")
|