Commit Graph

37 Commits (d8a51f34fd15dc4fbbe4e8416cd3270f23818c09)

Author SHA1 Message Date
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 433fabf7d8 Fix [24.0.6] Train toml config seed type error #2370 2024-04-25 13:10:55 -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 321325a0e8 Update release 2024-04-17 13:28:18 -04:00
bmaltais 8bda4f203c Align toml file content to sd-scripts defaults 2024-04-16 21:17:53 -04:00
bmaltais a22d4622cd
Add support for metadata parameters (#2295) 2024-04-15 13:26:39 -04:00
bmaltais b9eefb2370 Cleanup old commented code 2024-04-15 09:16:47 -04:00
bmaltais eec6f9baf4 Convert str numbers to proper int or float 2024-04-15 08:02:39 -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 a093b783bc
Fix accelerate issue on linux (#2281)
* Fix issue discovered after removing shell=True. This is a significant rewrite... but will make things better for the future.
2024-04-13 16:52:27 -04:00
bmaltais fd8bfa8a39 Upgrade Gradio release 2024-04-10 20:48:12 -04:00
bmaltais 63437a169b Update DB test config 2024-04-09 18:46:26 -04:00
bmaltais e8fa9178c4 Ad new caption tool 2024-04-06 09:15:55 -04:00
bmaltais ce1369d47b
Add support for Debiased Estimation loss to dreambooth (#2194) 2024-04-01 20:05:26 -04:00
bmaltais c4a11042f7 Update sd-script and add new test config 2024-04-01 11:30:54 -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 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
Bernard Maltais aaf0396d62 Fix vae path validation 2024-03-21 13:37:02 -04:00
bmaltais cdd16b22dc Adding support for LyCORIS BOFT, QLyCORIS and DoRA 2024-03-19 21:05:10 -04:00
bmaltais 99691b7b5d
Merge dev-toml (#2131)
* Integrate toml file option in GUI
2024-03-19 18:21:40 -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 1d61833c8a Fix issue with VAE list 2024-03-09 13:55:13 -05:00
bmaltais 3e023952df
Dev pure (#2039)
* Remove library folder, update setup_windows

* Cleanup sd-scripts files

* set sd-scripts release via file

* More cleanup

* Point to sd-scripts python files

* Fix blip captioning working directory

* Fix typo

* Fix issue with verify lora

* Improve git cloning of sd-scripts repo

* Integrate cloning in linux setup script

* Remove unnecessary folder

* Update linux setup

* Removing unnecessary code

* Fix validation issue under linux

* Update tensorboard version linux

* Updating linux requirements

* Update README

* Added submodule for sd-scripts

* Update for proper submodule handling

* Simplify submodule init

* Update comment

* Improve WSL2 LD_LIBRARY_PATH handling

* Update tensorboard handling of logging_dir

* chore(docker): update TensorFlow and TensorBoard versions

- Upgrade TensorFlow version in Dockerfile and requirements to 2.15.0.post1
- Update TensorBoard version in requirements to 2.15.2

Signed-off-by: 陳鈞 <jim60105@gmail.com>

* chore(docker): Add a mount for `sd-scripts` in the Dockerfile during pip install

- Add a mount step for `sd-scripts` in the Dockerfile during pip install.

Signed-off-by: 陳鈞 <jim60105@gmail.com>

* doc(docker): Update readme about docker

* Fix issue with TE LR

* Fix issue with TE LR

* Update sd-scripts to Feb 27 2024 commit

* Update version of bnb for windows

* Update windows setup

* Update windows setup

* Update bnb linux to match windows

* remove height:auto

* update UI

 * unify Quick Pick + custom model selection into gr.Dropdown()
 * add list_dirs(), list_files()
 * use gr.Accordion()
 * always show model + training image folder + output name + minimal default settings
 * do not show hidden files
 * clean up list_*()

* use PYTHONPATH env variable

* add "tool" classname to gr.Button() for sd-webui compatible

* update *_caption_gui.py using create_refresh_buttons()

* update Utilities->Group Images

* update Utilities->Convert model

* update dataset gui

* update Dreambooth/LoRA folders preparation

* update *lora*_gui UI using list_files(), create_refresh_button()

* more fixes for PYTHONPATH

* Minor edits to basic caption and common gui

* Optimising some code

* Fix path issue

* Fix issues with list_images_dir

* Update finetuning image folder label

* Minor updates to UI

---------

Signed-off-by: 陳鈞 <jim60105@gmail.com>
Co-authored-by: 陳鈞 <jim60105@gmail.com>
Co-authored-by: Won-Kyu Park <wkpark@gmail.com>
2024-03-09 09:30:20 -05:00
bmaltais fed7d7f997 Implement required GUI changes 2023-11-11 21:13:11 -05:00
Sam McLeod 72494a6687 fix: linting, spelling 2023-10-10 12:46:27 +11:00
bmaltais bc5973fefb Update print command to a regulat console string 2023-07-08 20:30:44 -04:00
bmaltais 5f6d0219aa Update presets 2023-07-08 19:49:04 -04:00
bmaltais 429d2f282f Update gradio release 2023-07-08 16:16:06 -04:00
bmaltais 72f3cb0f03 Making min and max steps parameters available on all trainers 2023-07-03 11:43:39 -04:00
bmaltais fae048ca55 Add support for now SDXL parameters 2023-07-03 10:04:55 -04:00
bmaltais cbb2134b4d Update Finetuning tab structure 2023-07-03 06:56:58 -04:00
bmaltais 8997825701 Various GUI updates 2023-07-02 15:19:09 -04:00
bmaltais be7c2cacaf Improvigs UI a bit 2023-07-02 13:30:48 -04:00
bmaltais 699ed33514 Add new test cases 2023-06-23 15:21:07 -04:00
bmaltais e1ad9ebef6 Update bitsandbytes 2023-06-23 13:48:43 -04:00
bmaltais d9c0648fdf Ass DB test 2023-06-21 09:10:16 -04:00