1580158
/
kohya_ss
mirror of https://github.com/bmaltais/kohya_ss
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
2,917 Commits
16 Branches
178 Tags
68 MiB
Commit Graph

60 Commits (415d7b25684070dba74059f06aa46d27e81e8a14)

Author SHA1 Message Date
bmaltais a97d3a94ee Fix issue with pre and post fix caption in subfolders 2024-04-29 07:13:34 -04:00
bmaltais d8a51f34fd Set `max_train_steps` to 0 if not specified in older `.json` config files 2024-04-26 07:11:07 -04:00
bmaltais 1f5f3cf027 Prevent tkinter import crash 2024-04-25 11:20:27 -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 5a801649c2 Convert config with use_wandb to log_with = wandb 2024-04-19 13:16:23 -04:00
bmaltais 542af98980 Fix wrong default for max_train_steps and max_train_epoch 2024-04-19 13:04:02 -04:00
bmaltais 4c5f4dfd27 update file read and write to utf-8 2024-04-19 10:50:43 -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
Bernard Maltais be5d30130f Fix clip_skip issue 2024-04-17 13:25:55 -04:00
bmaltais e6a8dec98d
Implement use_shell as parameter (#2297) 2024-04-15 20:26:09 -04:00
bmaltais a8320e3f2c Add support for accelerate dynamo parameters 2024-04-15 10:31:05 -04:00
bmaltais c4e0655a7a Fix accelerate path for linux systems 2024-04-15 08:50:48 -04:00
bmaltais eec6f9baf4 Convert str numbers to proper int or float 2024-04-15 08:02:39 -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
bmaltais fa5c65eade Update readme 2024-04-13 21:22:52 -04:00
bmaltais fddac3362e Make Stop training diseapear when training is done 2024-04-13 21:10:31 -04:00
bmaltais 4bfa486ed7 Improving args style 2024-04-13 16:46:40 -04:00
bmaltais 80afccc0e4 More fixes 2024-04-13 14:36:07 -04:00
bmaltais 1e7a263f79 Fix checkpoint name and sample prompt path 2024-04-13 14:23:46 -04:00
Bernard Maltais 835b6505c0 Fix running on linux 2024-04-13 14:03:55 -04:00
bmaltais 763eda2748 Update requirements for gradio 
More shlex.quote fix
 2024-04-13 12:21:26 -04:00
bmaltais 7581c4949f More fixes 2024-04-13 11:47:52 -04:00
bmaltais 62aa7ba8b6 Fix paths 2024-04-13 11:09:03 -04:00
bmaltais 79a78246a2 Rewrite run_cmd to list 2024-04-13 10:42:14 -04:00
bmaltais fef3dd2d12 Quick fix for other trainers 2024-04-13 08:43:57 -04:00
bmaltais f34eea41ca Fix issue for Dreambooth training... quick fix 2024-04-13 08:29:00 -04:00
bmaltais 25bb1303ff
Restrict extensions pre_postfix can use to improve security (#2259) 2024-04-11 08:55:03 -04:00
bmaltais 8bc67a7467
Find replace (#2258) 
* Improve security to limit finding and replacing contents of any files on the system, that the attacker knows the contents of by specifying the list of acceptable extensions for captions.

* Update caption extension fields to Dropdown with restricted choices for security reasons
 2024-04-11 08:32:17 -04:00
bmaltais 572dcd10ab Add Huber loss support 2024-04-07 10:35:07 -04:00
bmaltais 7b74dbfff8 Add code to check config file folder path existance 2024-04-05 07:21:17 -04:00
bmaltais ef1dcafbfe Fix warning messages 2024-04-03 20:20:59 -04:00
bmaltais 49f76343b5 Format with black 2024-04-03 18:32:25 -04:00
bmaltais c827268bf3 Cleanup old files and code 2024-04-03 18:17:59 -04:00
bmaltais b3a78febd1
Move Accelerate launch to own accordion (#2192) 
* Move accelerate launch parameters to own accordion
 2024-04-01 16:07:34 -04:00
Bernard Maltais ebc157bb68 Add support for Main process port 2024-03-29 20:56:37 -04:00
bmaltais 381e85296a Add support for masked_loss to the GUI 2024-03-26 20:14:07 -04:00
bmaltais b0139c751a Add gui support for save_state_on_train_end 2024-03-26 19:22:27 -04:00
bmaltais 0f889a8eb6 Add gui support for noise_offset_random_strength, ip_noise_gamma, ip_noise_gamma_random_strength 2024-03-26 19:12:48 -04:00
bmaltais 85c2b37a6a Fix max grad norm issue in DB 2024-03-25 10:54:51 -04:00
bmaltais 64360cec9e Fix issue where max_grad_norm = 0 would not show in the command line arguments 2024-03-24 20:46:38 -04:00
Bernard Maltais aaf0396d62 Fix vae path validation 2024-03-21 13:37:02 -04:00
bmaltais 99691b7b5d
Merge dev-toml (#2131) 
* Integrate toml file option in GUI
 2024-03-19 18:21:40 -04:00
bmaltais 5878bd6139 Move config for gui to seperate class 
Add support for more config paths
 2024-03-17 11:08:54 -04:00
bmaltais 6386a720b8 Add support for user managed path config 2024-03-16 20:58:22 -04:00
bmaltais 7186869c72
Code improvements (#2107) 
* Fix incorrect float representation

* Improve code

* Improve code quality

* Improve code

* Updates for https://github.com/bmaltais/kohya_ss/issues/2103
 2024-03-15 22:54:00 -04:00
bmaltais 90a02984cf Cleanup .update deprecations warnings 2024-03-15 10:03:41 -04:00
bmaltais 7081b22cba Add support for `wandb_run_name`, `log_tracker_name` and `log_tracker_config` parameters under the advanced section. 2024-03-14 19:41:34 -04:00
bmaltais d16be7b7c7 Update README, fix vae path validation 2024-03-13 07:11:41 -04:00