rockerBOO
9c591bdb12
Remove unnecessary subset line from collate
2023-11-05 16:58:20 -05:00
rockerBOO
3de9e6c443
Add validation split of datasets
2023-11-05 12:37:44 -05:00
rockerBOO
5b19bda85c
Add validation loss
2023-11-05 12:35:46 -05:00
Kohya S
6231aa91e2
common lr logging, set default None to ddp_timeout
2023-11-05 19:09:17 +09:00
Kohya S
a9ed4ed8a8
Merge pull request #900 from xzuyn/paged_adamw_32bit
...
Add PagedAdamW32bit
2023-10-29 15:01:55 +09:00
Kohya S
9d6a5a0c79
Merge pull request #899 from shirayu/use_moving_average
...
Show moving average loss in the progress bar
2023-10-29 14:37:58 +09:00
Kohya S
fb97a7aab1
Merge pull request #898 from shirayu/update_repare_buckets_latents
...
Fix a typo and add assertions in making buckets
2023-10-29 14:29:53 +09:00
Kohaku-Blueleaf
1cefb2a753
Better implementation for te autocast ( #895 )
...
* Better implementation for te
* Fix some misunderstanding
* as same as unet, add explicit convert
* Better cache TE and TE lr
* Fix with list
* Add timeout settings
* Fix arg style
2023-10-28 15:49:59 +09:00
xzuyn
d8f68674fb
Update train_util.py
2023-10-27 07:05:53 -04:00
Yuta Hayashibe
0d21925bdf
Use @property
2023-10-27 18:14:27 +09:00
Yuta Hayashibe
3d2bb1a8f1
Add LossRecorder and use moving average in all places
2023-10-27 17:49:49 +09:00
Yuta Hayashibe
837a4dddb8
Added assertions
2023-10-26 13:34:36 +09:00
青龍聖者@bdsqlsz
202f2c3292
Debias Estimation loss ( #889 )
...
* update for bnb 0.41.1
* fixed generate_controlnet_subsets_config for training
* Revert "update for bnb 0.41.1"
This reverts commit 70bd3612d8 .
* add debiased_estimation_loss
* add train_network
* Revert "add train_network"
This reverts commit 6539363c5c .
* Update train_network.py
2023-10-23 22:59:14 +09:00
Isotr0py
592014923f
Support JPEG-XL on windows
2023-10-04 21:48:25 +08:00
Yuta Hayashibe
27f9b6ffeb
updated typos to v1.16.15 and fix typos
2023-10-01 21:51:24 +09:00
Disty0
209eafb631
IPEX attention optimizations
2023-09-28 14:02:25 +03:00
Kohya S
624edf428f
Merge pull request #825 from Disty0/dev
...
Intel ARC support with IPEX
2023-09-24 14:29:03 +09:00
青龍聖者@bdsqlsz
d5be8125b0
update bitsandbytes for 0.41.1 and fixed bugs with generate_controlnet_subsets_config for training ( #823 )
...
* update for bnb 0.41.1
* fixed generate_controlnet_subsets_config for training
* Revert "update for bnb 0.41.1"
This reverts commit 70bd3612d8 .
2023-09-24 10:51:47 +09:00
Disty0
b99cd2a920
Update getDeviceIdListForCard
2023-09-20 17:16:06 +03:00
Disty0
b64389c8a9
Intel ARC support with IPEX
2023-09-19 18:05:05 +03:00
Kohya S
90c47140b8
add support model without position_ids
2023-09-13 17:59:34 +09:00
Kohya S
360af27749
fix ControlNetDataset not working
2023-09-03 12:27:58 +09:00
Kohya S
0ee75fd75d
fix typos, add comments etc.
2023-09-03 12:24:15 +09:00
Kohya S
2eae9b66d0
Merge pull request #798 from vvern999/vvern999-patch-1
...
add input perturbation noise
2023-09-03 10:51:23 +09:00
Kohya S
f6d417e26d
Merge pull request #791 from Isotr0py/dev
...
Intergrate fp16/bf16 support to sdxl model loading
2023-09-03 10:35:09 +09:00
Kohya S
948cf17499
add caption_prefix/suffix to dataset
2023-09-02 16:17:12 +09:00
Kohya S
497051c14b
Merge pull request #786 from Isotr0py/jxl
...
Support JPEG XL
2023-09-02 15:30:07 +09:00
Kohya S
c06a86706a
support diffusers' new VAE
2023-09-02 14:54:42 +09:00
vvern999
e0beb6a999
add input perturbation noise
...
from https://arxiv.org/abs/2301.11706
2023-09-02 07:33:27 +03:00
Kohya S
7e850f3b7e
Merge branch 'main' into sdxl
2023-09-01 07:59:26 +09:00
Isotr0py
2e0942d5c8
delet missed line
2023-08-27 20:45:40 +08:00
Isotr0py
6155f9c171
intergrate fp16/bf16 to model loading
2023-08-27 19:16:23 +08:00
Isotr0py
5d88351bb5
support jpeg xl
2023-08-25 11:07:02 +08:00
Kohya S
1161a5c6da
fix debug_dataset for controlnet dataset
2023-08-20 17:39:48 +09:00
Kohya S
e191892824
fix bucketing doesn't work in controlnet training
2023-08-20 12:24:40 +09:00
Kohya S
3f7235c36f
add lora controlnet train/gen temporarily
2023-08-17 10:08:02 +09:00
Kohya S
3307ccb2dc
revert default noise offset to 0 (None) in sdxl
2023-08-11 20:35:46 +09:00
Kohya S
6889ee2b85
add warning for bucket_reso_steps with SDXL
2023-08-11 19:02:36 +09:00
Kohya S
e73d103eca
fix sample gen failed in sdxl training
2023-08-11 16:58:52 +09:00
Kohya S
daad50e384
fix to work when input_ids has multiple EOS tokens
2023-08-10 20:13:59 +09:00
Kohya S
6f80fe17fc
fix crashing in saving lora with clipskip
2023-08-08 21:03:16 +09:00
Kohya S
78226f8574
change assert to print
2023-08-06 22:35:01 +09:00
Kohya S
3cdbbb43be
fix error in huggingface_path_in_repo=None
2023-08-06 22:08:30 +09:00
Kohya S
92f41f1051
update sdxl ver in lora metadata from v0-9 to v1-0
2023-08-06 22:06:48 +09:00
Kohya S
c142dadb46
support sai model spec
2023-08-06 21:50:05 +09:00
Kohya S
e5f9772a35
fix training textencoder in sdxl not working
2023-08-05 21:22:50 +09:00
reid3333
a02056c566
fix: load may fail if symbolic link points to relative path
2023-08-05 17:47:43 +09:00
Kohya S
9d855091bf
make bitsandbytes optional
2023-08-04 22:29:14 +09:00
Kohya S
f3be995c28
remove debug print
2023-08-04 08:44:17 +09:00
Kohya S
9d7619d1eb
remove debug print
2023-08-04 08:42:54 +09:00