Commit Graph

30 Commits

Author SHA1 Message Date
bmaltais
03bd2e9b01 Add TI training support 2023-01-26 16:22:58 -05:00
bmaltais
49bada0d25 Update default save model file format to safetensors 2023-01-22 18:21:09 -05:00
bmaltais
511361c80b - Add new tool to verify LoRA weights produced by the trainer. Can be found under "Dreambooth LoRA/Tools/Verify LoRA 2023-01-22 11:40:14 -05:00
bmaltais
2ca17f69dd v20.4.0:
Add support for `network_alpha` under the Training tab and support for `--training_comment` under the Folders tab.
2023-01-22 10:18:00 -05:00
bmaltais
fcad6bfd98 Add default for caption extension tools 2023-01-20 18:41:37 -05:00
bmaltais
31a1c8a71a Merge kohya Jan 19 updates 2023-01-19 15:47:43 -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
dc5afbb057 Move functions to common_gui
Add model name support
2023-01-09 11:48:57 -05:00
bmaltais
442eb7a292 Merge latest kohya code release into GUI repo 2023-01-09 07:47:07 -05:00
bmaltais
c79a8a9409 Add find/replace option to Basic Caption utility 2023-01-08 19:12:01 -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
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
ee2499d834 Merge latest kohya_ss code update 2023-01-01 13:10:32 -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
5e3f32f69c Update GUI 2022-12-22 20:18:51 -05:00
bmaltais
f60c8addd5 Making gui more modular by dividing it 2022-12-22 11:51:34 -05:00
bmaltais
6a7e27e100 fix issue with dataset balancing when the number of detected images in the folder is 0 2022-12-21 11:02:49 -05:00
bmaltais
1bc5089db8 Create model and log folder when running th dreambooth folder creation utility 2022-12-20 10:07:22 -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
1d412726b3 Add new prefix and postfic for captions 2022-12-19 11:43:29 -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