Plat
a823fd9fb8
Improve wandb logging ( #1576 )
...
* fix: wrong training steps were recorded to wandb, and no log was sent when logging_dir was not specified
* fix: checking of whether wandb is enabled
* feat: log images to wandb with their positive prompt as captions
* feat: logging sample images' caption for sd3 and flux
* fix: import wandb before use
2024-09-11 22:21:16 +09:00
Kohya S
c7c666b182
fix typo
2024-09-11 22:12:31 +09:00
Kohya S.
d83f2e92da
Merge pull request #1592 from cocktailpeanut/sd3
...
Critical typo fix
2024-09-11 22:10:37 +09:00
cocktailpeanut
8311e88225
typo fix
2024-09-11 09:02:29 -04:00
Kohya S
eaafa5c9da
Merge branch 'dev' into sd3
2024-09-11 21:46:21 +09:00
Kohya S
6dbfd47a59
Fix to work PIECEWISE_CONSTANT, update requirement.txt and README #1393
2024-09-11 21:44:36 +09:00
青龍聖者@bdsqlsz
fd68703f37
Add New lr scheduler ( #1393 )
...
* add new lr scheduler
* fix bugs and use num_cycles / 2
* Update requirements.txt
* add num_cycles for min lr
* keep PIECEWISE_CONSTANT
* allow use float with warmup or decay ratio.
* Update train_util.py
2024-09-11 21:25:45 +09:00
Kohya S
65b8a064f6
update README
2024-09-10 21:20:38 +09:00
Kohya S
d10ff62a78
support individual LR for CLIP-L/T5XXL
2024-09-10 20:32:09 +09:00
Kohya S
d29af146b8
add negative prompt for flux inference script
2024-09-09 23:01:15 +09:00
Kohya S
ce144476cf
Merge branch 'dev' into sd3
2024-09-07 10:59:22 +09:00
Kohya S
62ec3e6424
Merge branch 'main' into dev
2024-09-07 10:52:49 +09:00
Kohya S.
de25945a93
Merge pull request #1550 from kohya-ss/dependabot/github_actions/crate-ci/typos-1.24.3
...
Bump crate-ci/typos from 1.19.0 to 1.24.3
2024-09-07 10:50:46 +09:00
Kohya S
0005867ba5
update README, format code
2024-09-07 10:45:18 +09:00
Kohya S.
16bb5699ac
Merge pull request #1426 from sdbds/resize
...
Replacing CV2 resize to Pil resize
2024-09-07 10:22:52 +09:00
Kohya S.
319e4d9831
Merge pull request #1433 from millie-v/sample-image-without-cuda
...
Generate sample images without having CUDA (such as on Macs)
2024-09-07 10:19:55 +09:00
Kohya S
2889108d85
feat: Add --cpu_offload_checkpointing option to LoRA training
2024-09-05 20:58:33 +09:00
Kohya S
d9129522a6
set dtype before calling ae closes #1562
2024-09-05 12:20:07 +09:00
Kohya S
90ed2dfb52
feat: Add support for merging CLIP-L and T5XXL LoRA models
2024-09-05 08:39:29 +09:00
Kohya S
56cb2fc885
support T5XXL LoRA, reduce peak memory usage #1560
2024-09-04 23:15:27 +09:00
Kohya S
b7cff0a754
update README
2024-09-04 21:35:47 +09:00
Kohya S
b65ae9b439
T5XXL LoRA training, fp8 T5XXL support
2024-09-04 21:33:17 +09:00
Kohya S
6abacf04da
update README
2024-09-02 13:05:26 +09:00
Kohya S
4f6d915d15
update help and README
2024-09-01 19:12:29 +09:00
Kohya S.
1e30aa83b4
Merge pull request #1541 from sdbds/flux_shift
...
Add Flux_Shift for solving the problem of multi-resolution training blurry
2024-09-01 19:00:16 +09:00
Kohya S
92e7600cc2
Move freeze_blocks to sd3_train because it's only for sd3
2024-09-01 18:57:07 +09:00
青龍聖者@bdsqlsz
ef510b3cb9
Sd3 freeze x_block ( #1417 )
...
* Update sd3_train.py
* add freeze block lr
* Update train_util.py
* update
2024-09-01 18:41:01 +09:00
Kohya S.
928e0fc096
Merge pull request #1529 from Akegarasu/sd3
...
fix: text_encoder_conds referenced before assignment
2024-09-01 18:29:27 +09:00
dependabot[bot]
1bcf8d600b
Bump crate-ci/typos from 1.19.0 to 1.24.3
...
Bumps [crate-ci/typos](https://github.com/crate-ci/typos ) from 1.19.0 to 1.24.3.
- [Release notes](https://github.com/crate-ci/typos/releases )
- [Changelog](https://github.com/crate-ci/typos/blob/master/CHANGELOG.md )
- [Commits](https://github.com/crate-ci/typos/compare/v1.19.0...v1.24.3 )
---
updated-dependencies:
- dependency-name: crate-ci/typos
dependency-type: direct:production
update-type: version-update:semver-minor
...
Signed-off-by: dependabot[bot] <support@github.com >
2024-09-01 01:33:04 +00:00
Kohya S.
f8f5b16958
Merge pull request #1540 from kohya-ss/dependabot/pip/opencv-python-4.8.1.78
...
Bump opencv-python from 4.7.0.68 to 4.8.1.78
2024-08-31 21:37:07 +09:00
Kohya S.
826ab5ce2e
Merge pull request #1532 from nandometzger/main
...
Update train_util.py, bug fix
2024-08-31 21:36:33 +09:00
sdbds
25c9040f4f
Update flux_train_utils.py
2024-08-31 19:53:59 +08:00
dependabot[bot]
3a6154b7b0
Bump opencv-python from 4.7.0.68 to 4.8.1.78
...
Bumps [opencv-python](https://github.com/opencv/opencv-python ) from 4.7.0.68 to 4.8.1.78.
- [Release notes](https://github.com/opencv/opencv-python/releases )
- [Commits](https://github.com/opencv/opencv-python/commits )
---
updated-dependencies:
- dependency-name: opencv-python
dependency-type: direct:production
...
Signed-off-by: dependabot[bot] <support@github.com >
2024-08-31 06:21:16 +00:00
Nando Metzger
2a3aefb4e4
Update train_util.py, bug fix
2024-08-30 08:15:05 +02:00
Akegarasu
35882f8d5b
fix
2024-08-29 23:03:43 +08:00
Akegarasu
34f2315047
fix: text_encoder_conds referenced before assignment
2024-08-29 22:33:37 +08:00
Kohya S
8fdfd8c857
Update safetensors to version 0.4.4 in requirements.txt #1524
2024-08-29 22:26:29 +09:00
Kohya S
8ecf0fc4bf
Refactor code to ensure args.guidance_scale is always a float #1525
2024-08-29 22:10:57 +09:00
Kohya S.
930d709e3d
Merge pull request #1525 from Akegarasu/sd3
...
make guidance_scale keep float in args
2024-08-29 22:08:57 +09:00
Kohya S.
daa6ad5165
Update README.md
2024-08-29 21:25:30 +09:00
Kohya S
a0cfb0894c
Cleaned up README
2024-08-29 21:20:33 +09:00
Akegarasu
6c0e8a5a17
make guidance_scale keep float in args
2024-08-29 14:50:29 +08:00
Kohya S
a61cf73a5c
update readme
2024-08-27 21:44:10 +09:00
Kohya S
3be712e3e0
feat: Update direct loading fp8 ckpt for LoRA training
2024-08-27 21:40:02 +09:00
Kohya S
0087a46e14
FLUX.1 LoRA supports CLIP-L
2024-08-27 19:59:40 +09:00
Kohya S
72287d39c7
feat: Add shift option to --timestep_sampling in FLUX.1 fine-tuning and LoRA training
2024-08-25 16:01:24 +09:00
Kohya S
ea9242653c
Merge branch 'dev' into sd3
2024-08-24 21:24:44 +09:00
Kohya S
d5c076cf90
update readme
2024-08-24 21:21:39 +09:00
Kohya S.
4ca29edbff
Merge pull request #1505 from liesened/patch-2
...
Add v-pred support for SDXL train
2024-08-24 21:16:53 +09:00
Kohya S
5639c2adc0
fix typo
2024-08-24 16:37:49 +09:00
