Commit Graph

58 Commits

Author SHA1 Message Date
bmaltais
482d7834d1 Emergency fix 2023-03-02 17:51:17 -05:00
bmaltais
182080bb78 Run blue 2023-03-01 19:24:11 -05:00
bmaltais
7f0e5683c6 v21.0.1 2023-03-01 19:02:04 -05:00
bmaltais
1e3055c895 Update tensorboard 2023-03-01 13:14:47 -05:00
bmaltais
9d2e3f85a2 Add tensorboard support 2023-02-26 19:49:22 -05:00
bmaltais
f213b15014 Updates 2023-02-24 20:37:51 -05:00
bmaltais
bf0344ba9e Adding GUI support for new features 2023-02-22 20:32:57 -05:00
bmaltais
dfc9a8dd40 Fix issue with save config 2023-02-20 07:56:24 -05:00
bmaltais
758bfe85dc Adding support for Lion optimizer in gui 2023-02-19 20:13:03 -05:00
bmaltais
e5f8ba559f Add server_port and inbrowser support
- to all gui scripts
2023-02-10 08:22:03 -05:00
bmaltais
09d3a72cd8 Adding support for caption dropout 2023-02-07 20:58:35 -05:00
bmaltais
cbfc311687 Integrate new bucket parameters in GUI 2023-02-05 20:07:00 -05:00
bmaltais
2626214f8a Add support for LoRA resizing 2023-02-04 11:55:06 -05:00
bmaltais
202923b3ce Add support for --keep_token option 2023-01-27 07:33:44 -05:00
bmaltais
f265197ba9 Remove unneeded arguments 2023-01-19 20:51:40 -05:00
bmaltais
567221549f Complete training code refactoring 2023-01-15 19:59:40 -05:00
bmaltais
123cf4e3c5 Move cxommon adv train parm to common gui 2023-01-15 15:03:04 -05:00
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