Commit Graph

291 Commits

Author SHA1 Message Date
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
42a3646d4a Update readme 2023-01-09 17:59:11 -05:00
bmaltais
11fbc63440 Fix issue with lora model loading 2023-01-09 17:22:42 -05:00
bmaltais
f1b595a141
Merge pull request #35 from bmaltais/dev
Add support for --output_name to GUI
2023-01-09 11:49:53 -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
442eb7a292 Merge latest kohya code release into GUI repo 2023-01-09 07:47:07 -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
c5c298201c
Merge pull request #34 from bmaltais/dev
Add new features
2023-01-08 19:32:21 -05:00
bmaltais
f1d53ae3f9 Add resume training and save_state option to finetune UI 2023-01-08 19:31:44 -05:00
bmaltais
c79a8a9409 Add find/replace option to Basic Caption utility 2023-01-08 19:12:01 -05:00
bmaltais
4f43e3763a
Merge pull request #32 from bmaltais/dev
Emergency fix
2023-01-06 23:20:14 -05:00
bmaltais
115ed35187 Emergency fix 2023-01-06 23:19:49 -05:00
bmaltais
1888e0d539
Merge pull request #31 from bmaltais/dev
Update readme
2023-01-06 18:38:46 -05:00
bmaltais
aa0e39e14e Update readme 2023-01-06 18:38:24 -05:00
bmaltais
3dd8926585
Merge pull request #30 from bmaltais/dev
Update README
2023-01-06 18:37:20 -05:00
bmaltais
8ec1edbadc Update README 2023-01-06 18:33:07 -05:00
bmaltais
f2ba97a03e
Merge pull request #29 from bmaltais/dev
Add new Utility to Extract a LoRA from a finetuned model
2023-01-06 18:26:18 -05:00
bmaltais
34f7cd8e57 Add new Utility to Extract a LoRA from a finetuned model 2023-01-06 18:25:55 -05:00
bmaltais
4ad98ddea7
Merge pull request #27 from bmaltais/dev
Emergency fix for dreambooth_ui no longer working, sorry
2023-01-06 07:13:39 -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
7e6677b5f6
Merge pull request #26 from bmaltais/dev
- Add support for `--clip_skip` option
2023-01-05 19:16:52 -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
0251bf3064
Merge pull request #24 from bmaltais/dev
Add multiple options to fine_tune.py
2023-01-02 13:08:04 -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
1a9554a800
Merge pull request #23 from bmaltais/dev
Quick fix for bug when no LoRA network weights is provided
2023-01-01 14:38:07 -05:00
bmaltais
bfa590b313 Quick fix for bug when no LoRA network weights is provided 2023-01-01 14:37:32 -05:00
bmaltais
98f7dd5080
Merge pull request #22 from bmaltais/dev
Update LoRA GUI
2023-01-01 14:15:41 -05:00
bmaltais
af46ce4c47 Update LoRA GUI
Various improvements
2023-01-01 14:14:58 -05:00
bmaltais
ee2499d834 Merge latest kohya_ss code update 2023-01-01 13:10:32 -05:00
bmaltais
2c069d1067 Improve GUI 2023-01-01 00:33:29 -05:00
bmaltais
6b41f4f951 Fix standalone lora gui issue 2022-12-30 21:18:50 -05:00
bmaltais
4216d05471 set default dataset_repeats to 1, resolution check 2022-12-30 21:01:33 -05:00
bmaltais
cdf84e2f4b
Merge pull request #21 from bmaltais/dev
Add LoRA support to create custom networks
2022-12-30 20:58:40 -05:00
bmaltais
0f75b7c2db Update readme 2022-12-30 20:50:01 -05:00
bmaltais
0c5966f6f4 Fix issue with LR 2022-12-29 21:17:41 -05:00
bmaltais
b44f075f60 Implement open and save config for LoRA 2022-12-29 14:00:02 -05:00
bmaltais
0f42ab78c4 Fix issue with caption extension 2022-12-28 23:19:18 -05:00
bmaltais
d6def2343f !st rough implementation of LoRA GUI 2022-12-28 23:02:35 -05:00
Bernard Maltais
fa91c10f8c Add documentation 2022-12-26 09:17:10 -05:00
Bernard Maltais
39baddf805 Add LoRA support from sd_script repo 2022-12-26 08:47:33 -05:00
bmaltais
2d58594daa
Merge pull request #16 from bmaltais/dev
v18.8
2022-12-23 07:57:23 -05:00
bmaltais
2cdf4cf741 - Fix for conversion tool issue when the source was an sd1.x diffuser model
- Other minor code and GUI fix
2022-12-23 07:56:35 -05:00
bmaltais
5e3f32f69c Update GUI 2022-12-22 20:18:51 -05:00
bmaltais
89d275dff7
Merge dreambooth and Finetune is a common GUI
Merge dreambooth and Finetune is a common GUI
2022-12-22 13:20:00 -05:00
bmaltais
fd10512bf4 Add revision info 2022-12-22 13:19:28 -05:00
bmaltais
8a99af9269 Update GUI and Readme 2022-12-22 13:14:46 -05:00