Commit Graph

91 Commits (8fb9001a2c866a6796c8a34f3e7bb9d2d99ddb6a)

Author SHA1 Message Date
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 b77da103c4 Fix typo 2024-10-26 19:47:25 -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
rohitanshu b8c4de17e5
Update textual_inversion_gui.py
Fixed minor typos of "Regularization"
2024-09-16 19:45:41 +05:30
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 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 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
Krithic Kumar 3902707d9b fix: vae path validation 2024-05-07 13:41:24 +05:30
bmaltais 8f8907f36c Fix issue with vae file path validation 2024-05-06 06:43:43 -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 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
bmaltais aa4d13fc93 Merge branch 'master' of https://github.com/bmaltais/kohya_ss into dev 2024-04-17 07:20:50 -04:00
bmaltais 43087f7b3d Update training info display 2024-04-17 07:11:40 -04:00
bmaltais 8bda4f203c Align toml file content to sd-scripts defaults 2024-04-16 21:17:53 -04:00
bmaltais 13b836ad76 Fix issues with steps calculations 2024-04-16 14:52:20 -04:00
bmaltais 9037fd0373
Improve process execution under Linux (and windows) (#2301)
* Update how process is executed

* Auto config use_shell=True on posix systems
2024-04-16 07:53:44 -04:00
bmaltais e6a8dec98d
Implement use_shell as parameter (#2297) 2024-04-15 20:26:09 -04:00
bmaltais a22d4622cd
Add support for metadata parameters (#2295) 2024-04-15 13:26:39 -04:00
bmaltais a8320e3f2c Add support for accelerate dynamo parameters 2024-04-15 10:31:05 -04:00
bmaltais b9eefb2370 Cleanup old commented code 2024-04-15 09:16:47 -04:00
bmaltais c4e0655a7a Fix accelerate path for linux systems 2024-04-15 08:50:48 -04:00
bmaltais 18ef5305d2 Make gradient_accumulation_steps an int 2024-04-15 08:10:24 -04:00
bmaltais a08c7c6ef1 Update parameters handling 2024-04-14 19:10:26 -04:00
bmaltais e98fc40a63
- Implement HuggingFace inputs in all training tabs (#2287)
- Switch to sd-scripts configuration files to pass most of the parameters in a secure way.
2024-04-14 18:43:45 -04:00