sd-scripts (original) (raw)

and others added 4 commits

March 28, 2026 19:25

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 6d3e514.

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

train_util.py/deepspeed_utils.py の getattr 化を元に戻し、LECO パーサーにダミー引数を追加
sdxl_train_util のモジュールレベルインポートをローカルインポートに変更
PromptEmbedsCache.getitem でキャッシュミス時に KeyError を送出するよう修正
設定ファイル形式を YAML から TOML に変更（リポジトリの規約に統一）
重複コード (build_network_kwargs, get_save_extension, save_weights) を leco_train_util.py に統合
_expand_slider_target の冗長な PromptSettings 構築を簡素化
add_time_ids 用に専用の batch_add_time_ids 関数を追加

Co-Authored-By: Claude Opus 4.6 noreply@anthropic.com

コマンドライン引数の全カテゴリ別解説、プロンプト TOML の全フィールド説明、 2つの guidance_scale の違い、推奨設定表、YAML からの変換ガイド等を追加。英語本文と日本語折り畳みの二言語構成。

Co-Authored-By: Claude Opus 4.6 noreply@anthropic.com

torch.randn のデフォルト float32 により latents が暗黙的にアップキャストされる問題を修正。 float32/CPU で生成後に latents の dtype/device へ変換する安全なパターンを採用。

Co-Authored-By: Claude Opus 4.6 noreply@anthropic.com

This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.Learn more about bidirectional Unicode characters

[ Show hidden characters]({{ revealButtonHref }})