* Refactor and Improve Core Components
This commit includes several improvements to the core components of the Kohya_ss GUI:
1. **Refactored `check_torch()` in `setup/setup_common.py`**:
* Broke down the function into smaller, more focused functions for better readability and maintainability.
* Simplified the logic for detecting different hardware backends (CUDA, ROCm, Intel OneAPI).
2. **Improved Error Handling in `setup/setup_common.py`**:
* Made error messages more user-friendly and informative by providing context and suggestions.
* Ensured consistent logging of errors.
* Enhanced error reporting for external processes like Git and pip/uv.
3. **Enhanced Documentation for `config.toml`**:
* Updated the "Custom Path Defaults" section in `README.md` with more details on customizing `config.toml`.
* Provided clearer examples of common configurations.
4. **Reviewed and Updated Dependencies**:
* Removed several unused dependencies from `pyproject.toml` (`altair`, `fairscale`, `imagesize`, `invisible-watermark`, `pytorch-lightning`, `voluptuous`).
* Updated various dependencies to their latest stable and compatible versions.
---------
Co-authored-by: google-labs-jules[bot] <161369871+google-labs-jules[bot]@users.noreply.github.com>
- Fixes a typo in the variable name and log message from "reccomended" to the correct "recommended" when logging Apple MPS memory information.
Signed-off-by: CHEN, CHUN <jim60105@gmail.com>
* Adding some changes to support current apple silicone
Adding a note that MPS is detected in validation, and a current set of packages that offer MPS torch acceleration
* Adding MPS support for blip2
* 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
* 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>
bmaltais
CHEN, CHUN
Ryan
DukeSniper
Disty0
Disty0
Montri Ngamakharakul
Won-Kyu Park
Hina Chen
gabriel dhimoila
Sam McLeod
Enes Sadık Özbek