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