Spaces:
Running
Running
Update flux_train_ui.py
Browse files- flux_train_ui.py +1 -1
flux_train_ui.py
CHANGED
|
@@ -406,7 +406,7 @@ with gr.Blocks(theme=theme, css=css) as demo:
|
|
| 406 |
outputs=[captioning_area, sample, start]
|
| 407 |
)
|
| 408 |
|
| 409 |
-
start.click(fn=create_dataset, inputs=[images]
|
| 410 |
fn=start_training,
|
| 411 |
inputs=[
|
| 412 |
lora_name,
|
|
|
|
| 406 |
outputs=[captioning_area, sample, start]
|
| 407 |
)
|
| 408 |
|
| 409 |
+
start.click(fn=create_dataset, inputs=[images], outputs=dataset_folder).then(
|
| 410 |
fn=start_training,
|
| 411 |
inputs=[
|
| 412 |
lora_name,
|