Commit Graph

41 Commits

Author SHA1 Message Date
bmaltais
abccecb093 Move save_state and resume to common gui
Format code
2023-01-15 12:01:59 -05:00
bmaltais
6aed2bb402 Add support for new arguments:
- max_train_epochs
- max_data_loader_n_workers
Move some of the codeto  common gui library.
2023-01-15 11:05:22 -05:00
bmaltais
43116feda8 Add support for max token 2023-01-10 09:38:32 -05:00
bmaltais
dc5afbb057 Move functions to common_gui
Add model name support
2023-01-09 11:48:57 -05:00
bmaltais
fdb4508a62 - Add model output name parameter to dreambooth
- Simplify config load and save
- blue code
2023-01-09 10:14:40 -05:00
bmaltais
402cb51ec0 refactor Dreambooth gui code 2023-01-09 08:08:47 -05:00
bmaltais
a4262c0a66 - Add vae support to dreambooth GUI
- Add gradient_checkpointing, gradient_accumulation_steps, mem_eff_attn, shuffle_caption to finetune GUI
- Add gradient_accumulation_steps, mem_eff_attn to dreambooth lora gui
2023-01-08 20:55:41 -05:00
bmaltais
34f7cd8e57 Add new Utility to Extract a LoRA from a finetuned model 2023-01-06 18:25:55 -05:00
bmaltais
c20a10d7fd Emergency fix for dreambooth_ui no longer working, sorry
- Add LoRA network merge too GUI. Run `pip install -U -r requirements.txt` after pulling this new release.
2023-01-06 07:13:12 -05:00
bmaltais
b8100b1a0a - Add support for --clip_skip option
- Add missing `detect_face_rotate.py` to tools folder
- Add `gui.cmd` for easy start of GUI
2023-01-05 19:16:13 -05:00
bmaltais
9d3c402973 - Finetune, add xformers, 8bit adam, min bucket, max bucket, batch size and flip augmentation support for dataset preparation
- Finetune, add "Dataset preparation" tab to group task specific options
2023-01-02 13:07:17 -05:00
bmaltais
1d460a09fd add support for color and flip augmentation to "Dreambooth LoRA" 2023-01-01 22:43:44 -05:00
bmaltais
af46ce4c47 Update LoRA GUI
Various improvements
2023-01-01 14:14:58 -05:00
bmaltais
5e3f32f69c Update GUI 2022-12-22 20:18:51 -05:00
bmaltais
8a99af9269 Update GUI and Readme 2022-12-22 13:14:46 -05:00
bmaltais
f60c8addd5 Making gui more modular by dividing it 2022-12-22 11:51:34 -05:00
bmaltais
aa5d13f9a7 Add authentication support 2022-12-21 09:05:06 -05:00
bmaltais
706dfe157f
Merge dreambooth and finetuning in one repo to align with kohya_ss new repo (#10)
* Merge both dreambooth and finetune back in one repo
2022-12-20 09:15:17 -05:00
bmaltais
1f1dd5c4de Add support for:
shuffle_caption,
save_state,
resume,
prior_loss_weight,

Fix issue with config open and save
2022-12-19 21:50:05 -05:00
bmaltais
6987f51b0a Fix stop encoder training issue 2022-12-19 10:39:04 -05:00
bmaltais
61a9481903 Fix save as issue 2022-12-19 09:47:35 -05:00
bmaltais
c90aa2cc61 - Fix file/folder opening behind the browser window
- Add WD14 and BLIP captioning to utilities
- Improve overall GUI layout
2022-12-19 09:22:52 -05:00
bmaltais
0ca93a7aa7 v18.1: Model conversion utility 2022-12-18 13:11:10 -05:00
bmaltais
f459c32a3e v18: Save model as option added 2022-12-17 20:36:31 -05:00
bmaltais
e1d66e47f4
v17.2 (#8)
* Update youtube video

* Dataset balancing

* Fix typo
2022-12-17 16:22:34 -05:00
bmaltais
9b2ff7a8f8 Fix logic issue for reg images folder creation 2022-12-17 09:51:30 -05:00
bmaltais
42a0732e0d Reformat code with blue 2022-12-16 20:26:26 -05:00
bmaltais
90dad5471c Refactor GUI code to be modular 2022-12-16 13:16:23 -05:00
bmaltais
9bddf2ce4c Add support for image captionning 2022-12-16 11:04:06 -05:00
bmaltais
2cd356a37e Fix UI text 2022-12-15 19:32:53 -05:00
bmaltais
dddcffbe51 Update UI code 2022-12-15 19:04:26 -05:00
bmaltais
a7cd798abf Fix UI logic 2022-12-15 18:44:02 -05:00
bmaltais
2f91fd24ae Update Utilities logic 2022-12-15 18:34:54 -05:00
bmaltais
515c52e2cf Fix Utilities layout 2022-12-15 18:31:36 -05:00
bmaltais
cb98b4f149 Add new Utilities Tab 2022-12-15 18:19:35 -05:00
bmaltais
969eea519f Remove redundant code 2022-12-15 07:48:29 -05:00
bmaltais
3834e5dbab Adding improved elements to GUI 2022-12-14 14:40:24 -05:00
bmaltais
469b15b579 Adding support for new parameters 2022-12-13 21:21:59 -05:00
bmaltais
71765ae243 Fix steps issue when regularisation images are used 2022-12-13 17:52:25 -05:00
bmaltais
416912eee1 Refactor gui code 2022-12-13 14:59:33 -05:00
bmaltais
5f1a465a45 Update to v17
New GUI
2022-12-13 13:49:14 -05:00