Commit Graph

111 Commits (8fb9001a2c866a6796c8a34f3e7bb9d2d99ddb6a)

Author SHA1 Message Date
bmaltais 3c860c4035 Make blocks_to_swap common 2024-12-31 08:11:42 -05:00
bmaltais 7068c7dabe Add support for fused_backward_pass for sd3 finetuning 2024-12-30 19:37:03 -05:00
bmaltais fce89ad798 Add support for huber_scale 2024-12-30 19:21:03 -05:00
bmaltais d0cd9f547a Reverting all changes for max_train_steps 2024-11-18 19:31:51 -05:00
bmaltais 309a9bbc1b Fix max_train_steps_info error 2024-11-18 07:07:33 -05:00
bmaltais a6f0ff72a8 Fix issue with max_train_steps 2024-11-17 18:37:27 -05:00
bmaltais 021c6f5ae3 Add support for save_last_n_epochs 2024-10-29 20:14:02 -04:00
bmaltais b9fe58cd1e Add support for --save_last_n_epochs_state 2024-10-18 17:53:06 -04:00
bmaltais 3c3f1edfb6 Add --skip_cache_check option to GUI 2024-10-14 18:32:09 -04:00
bmaltais a28026bca1 Adding support for blocks_to_swap option to gui 2024-09-29 08:36:05 -04:00
bmaltais 20d7e659e1 Update lr_warmup_steps code 2024-09-12 21:31:18 -04:00
bmaltais e655ebf2dd Add support for lr_warmup_steps override 2024-09-12 20:56:05 -04:00
bmaltais 5668ff07a9 Add flux_fused_backward_pass to dreambooth and finetuning 2024-08-21 09:24:21 -04:00
bmaltais 33937c2284 Fix issue with split_mode and train_blocks 2024-08-21 07:52:46 -04:00
bmaltais f57973fbc5 Add mem_eff_save option to GUI for Flux.1 2024-08-19 10:35:25 -04:00
bmaltais 215a05388b Fix blockwise_fused_optimizers typo 2024-08-19 10:16:22 -04:00
bmaltais 8aa658f7f0 Temporary fir for blockwise_fused_optimizers 2024-08-18 17:45:10 -04:00
bmaltais 2611de717b add missing fp8_base parameter 2024-08-18 11:58:08 -04:00
bmaltais 647295aa88 Add dreambooth and finetuning support for flux.1 2024-08-18 11:29:15 -04:00
bmaltais 2f8749c40e Fix text encoder issue 2024-07-08 20:34:14 -04:00
bmaltais 37523a9f38 MVP GUI for SD3 2024-07-07 15:45:49 -04:00
bmaltais 65ac2f163a Adding SD3 GUI elements 2024-07-07 15:14:53 -04:00
bmaltais 441721ab87 Use output model name as wandb run name if not provided 2024-06-24 10:26:54 -04:00
bmaltais 848f4d45a8 Add support for custom learning rate scheduler type to the GUI 2024-05-25 08:21:25 -04:00
bmaltais ec5397f4d9 Add --log_config support to GUI 2024-05-24 20:07:33 -04:00
bmaltais 0b32f9cf05 Implement support for disable_mmap_load_safetensors 2024-05-12 20:09:00 -04:00
bmaltais 0d999a354a Implement Fused backward pass and fused optimiser groups 2024-05-12 19:55:05 -04:00
bmaltais 3b771f51ec
Fix finetuning meta file creation (#2488) 2024-05-11 09:10:39 -04:00
bmaltais d010b0d15b Fix resume folder path validation 2024-05-10 20:16:39 -04:00
bmaltais 73822af880 Catch when accelerate command is not found 2024-05-09 18:01:05 -04:00
bmaltais 27b58a79f2 Fix creation of output folders 2024-05-09 13:30:27 -04:00
bmaltais 92a01a3890 Fix issue with tensorboard 2024-05-09 13:20:20 -04:00
bmaltais 7fac1e89dd Move env var config to common_gui 2024-05-07 07:58:11 -04:00
bmaltais d26dad20b6
Relocate toml training config file to same folder as the model output directory (#2448) 2024-05-05 13:43:56 -04:00
bmaltais 36071cc244
Improve files and folders validation (#2429) 2024-05-01 08:29:31 -04:00
bmaltais 7bbc99d91b
2382 some incorrect usage related to the recent shell=false issue (#2417)
* Convert script back to no shell.

* Update rest of tools and update dreambooth to no shell

* Update rest of trainers to not use shell

* Allow the use of custom caption extension
2024-04-29 07:44:55 -04:00
bmaltais ee725a9c01 Change tmp file config name to have date and time info 2024-04-27 21:01:56 -04:00
bmaltais 433fabf7d8 Fix [24.0.6] Train toml config seed type error #2370 2024-04-25 13:10:55 -04:00
bmaltais 8234e52ded
Add validation of lr scheduler and optimizer arguments (#2358) 2024-04-20 15:53:11 -04:00
bmaltais 58e57a3365
Fix issue with lora_network_weights not being loaded (#2357) 2024-04-20 15:05:35 -04:00
bmaltais 4923f5647c
Make Start/Stop buttons visible in headless (#2356) 2024-04-20 14:27:19 -04:00
Maatra 25d7c6889f
Changed logger checkbox to dropdown, renamed use_wandb -> log_with (#2352) 2024-04-19 13:14:15 -04:00
bmaltais 4c5f4dfd27 update file read and write to utf-8 2024-04-19 10:50:43 -04:00
bmaltais 4df07c4a73 Write toml as UTF-8 2024-04-18 22:34:06 -04:00
bmaltais b4f1589f18
2330 another seemingly new data validation leads to unusable configs 2403 (#2339)
* Fix issue with None values staying as 0 in toml config file

* Fix noise_offset issues (and possibly other issues)
2024-04-18 21:47:00 -04:00
bmaltais 07db163be9
2325 quotes wrapping python path cause subprocess cant find target in v2403 (#2338)
* Fix issue with doublequotes wrapped python path

* Fix path. Always assume no space in path to kohya_ss repo

* Add img path with space

* Enforce no white space in path to kohya_ss
2024-04-18 20:45:43 -04:00
bmaltais 85bdb37789 Improve PYTHON variable 2024-04-17 20:09:06 -04:00
bmaltais 84fb7ce8c9 Update utility paths and print toml file when clicking the Print training command button 2024-04-17 19:54:56 -04:00
bmaltais 962fc681cc Fix issue with sample prompt creation 2024-04-17 17:45:59 -04:00
bmaltais 1898851aa6 Fix bad merge 2024-04-17 07:32:41 -04:00