Commit Graph

17 Commits (d8a51f34fd15dc4fbbe4e8416cd3270f23818c09)

Author SHA1 Message Date
bmaltais 542af98980 Fix wrong default for max_train_steps and max_train_epoch 2024-04-19 13:04:02 -04:00
bmaltais 8bda4f203c Align toml file content to sd-scripts defaults 2024-04-16 21:17:53 -04:00
bmaltais e519ffec06 Fix warning about dropdown 2024-04-15 13:59:13 -04:00
bmaltais b64d856340
Implement LR Warmup disabling when constant scheduler is used (#2292)
* Implement LR Warmup disabling when constant scheduler is used
2024-04-15 08:37:16 -04:00
bmaltais eec6f9baf4 Convert str numbers to proper int or float 2024-04-15 08:02:39 -04:00
bmaltais 5316db3fa7 Add huggingface parms to config file 2024-04-14 22:03:36 -04:00
bmaltais 91b679f9ca Update dreambooth attributes 2024-04-14 21:41:04 -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 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 fd8bfa8a39 Upgrade Gradio release 2024-04-10 20:48:12 -04:00
bmaltais 5bd439f3d0 Add basic section to config example.toml 2024-04-03 22:46:44 -04:00
bmaltais 49f76343b5 Format with black 2024-04-03 18:32:25 -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
bmaltais 6150a0f056 Update basic training code and layout 2024-03-16 11:37:36 -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 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
Won-Kyu Park bdb98af295
move gui related files to kohya_gui/ 2024-02-27 21:54:38 +09:00