Add Gradio UI for ai-toolkit (ostris#141)

* Add Gradio UI for FLUX.1 * small text changes * no flash-attn? no problem! * bye flash-attn! * fixes for windows --------- Co-authored-by: multimodalart <[email protected]>
NBSTpeterhill · Aug 30, 2024 · f84bd6d · f84bd6d
1 parent 60232de
commit f84bd6d
Show file tree

Hide file tree

Showing 4 changed files with 431 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -53,6 +53,20 @@ pip install -r requirements.txt
 
 ## FLUX.1 Training
 
+### Gradio UI
+
+To get started training locally with a with a custom UI, once you followed the steps above and `ai-toolkit` is installed:
+
+```bash
+cd ai-toolkit #in case you are not yet in the ai-toolkit folder
+huggingface-cli login #provide a `write` token to publish your LoRA at the end
+python flux_train_ui.py
+```
+
+You will instantiate a UI that will let you upload your images, caption them, train and publish your LoRA
+![image](assets/lora_ease_ui.png)
+
+
 ### Tutorial
 
 To get started quickly, check out [@araminta_k](https://x.com/araminta_k) tutorial on [Finetuning Flux Dev on a 3090](https://www.youtube.com/watch?v=HzGW_Kyermg) with 24GB VRAM.

diff --git a/assets/lora_ease_ui.png b/assets/lora_ease_ui.png