Kohya S.
|
5cdad10de5
|
Fix/leco cleanup (#2294)
* feat: SD1.x/2.x と SDXL 向けの LECO 学習スクリプトを追加 (#2285)
* 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>
* fix: format
* fix: LECO PR #2285 のレビュー指摘事項を修正
- 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>
* docs: LECO 学習ガイドを大幅に拡充
コマンドライン引数の全カテゴリ別解説、プロンプト TOML の全フィールド説明、
2つの guidance_scale の違い、推奨設定表、YAML からの変換ガイド等を追加。
英語本文と日本語折り畳みの二言語構成。
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
* fix: apply_noise_offset の dtype 不一致を修正
torch.randn のデフォルト float32 により latents が暗黙的にアップキャストされる問題を修正。
float32/CPU で生成後に latents の dtype/device へ変換する安全なパターンを採用。
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
---------
Co-authored-by: Umisetokikaze <52318966+umisetokikaze@users.noreply.github.com>
Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
|
2026-03-29 20:41:43 +09:00 |
|
Kohya S
|
c4be615f69
|
fix(tests): add ip_noise_gamma args for MockArgs in pytest
|
2026-01-18 15:05:57 +09:00 |
|
rockerBOO
|
c149cf283b
|
Add parser args for other trainers.
|
2025-08-03 00:58:25 -04:00 |
|
rockerBOO
|
056472c2fc
|
Add tests
|
2025-08-02 21:16:56 -04:00 |
|
Kohya S
|
6731d8a57f
|
fix: update system prompt handling
|
2025-06-29 22:21:48 +09:00 |
|
Kohya S
|
5034c6f813
|
feat: add workaround for 'gated repo' error on github actions
|
2025-06-29 22:00:58 +09:00 |
|
Kohya S
|
935e0037dc
|
feat: update lumina system prompt handling
|
2025-06-29 21:33:09 +09:00 |
|
rockerBOO
|
d94bed645a
|
Add lumina tests and fix image masks
|
2025-06-09 21:14:51 -04:00 |
|
rockerBOO
|
89f0d27a59
|
Set sigmoid_scale to default 1.0
|
2025-03-20 15:10:33 -04:00 |
|
rockerBOO
|
d40f5b1e4e
|
Revert "Scale sigmoid to default 1.0"
This reverts commit 8aa126582e.
|
2025-03-20 15:09:50 -04:00 |
|
rockerBOO
|
8aa126582e
|
Scale sigmoid to default 1.0
|
2025-03-20 15:09:11 -04:00 |
|
rockerBOO
|
e8b3254858
|
Add flux_train_utils tests for get get_noisy_model_input_and_timesteps
|
2025-03-20 15:01:15 -04:00 |
|