mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-15 16:39:42 +00:00
4ea6032c66
* Add LECO training script and associated tests
- Implemented `sdxl_train_leco.py` for training with LECO prompts, including argument parsing, model setup, training loop, and weight saving functionality.
- Created unit tests for `load_prompt_settings` in `test_leco_train_util.py` to validate loading of prompt configurations in both original and slider formats.
- Added basic syntax tests for `train_leco.py` and `sdxl_train_leco.py` to ensure modules are importable.
* fix: use getattr for safe attribute access in argument verification
* feat: add CUDA device compatibility validation and corresponding tests
* Revert "feat: add CUDA device compatibility validation and corresponding tests"
This reverts commit 6d3e51431b.
* feat: update predict_noise_xl to use vector embedding from add_time_ids
* feat: implement checkpointing in predict_noise and predict_noise_xl functions
* feat: remove unused submodules and update .gitignore to exclude .codex-tmp
---------
Co-authored-by: Kohya S. <52813779+kohya-ss@users.noreply.github.com>
16 lines
415 B
Python
16 lines
415 B
Python
import train_leco
|
|
from library import deepspeed_utils, train_util
|
|
|
|
|
|
def test_syntax():
|
|
assert train_leco is not None
|
|
|
|
|
|
def test_setup_parser_supports_shared_training_validation():
|
|
args = train_leco.setup_parser().parse_args(["--prompts_file", "slider.yaml"])
|
|
|
|
train_util.verify_training_args(args)
|
|
|
|
assert args.min_snr_gamma is None
|
|
assert deepspeed_utils.prepare_deepspeed_plugin(args) is None
|