Kohya S
2186e417ba
fix size of bucket < min_size ref #1008
2023-12-20 22:12:21 +09:00
Cauldrath
f8360a4831
Fix zero height buckets
...
If max_size is too large relative to max_reso, it will calculate a height of zero for some buckets.
This causes a crash later when it divides the width by the height.
This change also simplifies some math and consolidates the redundant "size" variable into "width".
2023-12-19 18:35:09 -05:00
Disty0
b64389c8a9
Intel ARC support with IPEX
2023-09-19 18:05:05 +03:00
Kohya S
c142dadb46
support sai model spec
2023-08-06 21:50:05 +09:00
Kohya S
4072f723c1
Merge branch 'main' into sdxl
2023-07-29 14:55:03 +09:00
Kohya S
1e4512b2c8
support ckpt without position id in sd v1 #687
2023-07-29 14:19:25 +09:00
Kohya S
747af145ed
add sdxl fine-tuning and LoRA
2023-06-26 08:07:24 +09:00
Kohya S
92e50133f8
Merge branch 'original-u-net' into dev
2023-06-17 21:57:08 +09:00
Kohya S
449ad7502c
use original unet for HF models, don't download TE
2023-06-14 22:26:05 +09:00
Kohya S
67f09b7d7e
change ver no for Diffusers VAE changing
2023-06-12 12:29:44 +09:00
ykume
4d0c06e397
support both 0.10.2 and 0.17.0 for Diffusers
2023-06-11 18:54:50 +09:00
ykume
4b7b3bc04a
fix saved SD dict is invalid for VAE
2023-06-11 17:35:00 +09:00
ykume
4e25c8f78e
fix to work with Diffusers 0.17.0
2023-06-11 16:57:17 +09:00
Kohya S
5db792b10b
initial commit for original U-Net
2023-06-03 19:24:47 +09:00
ddPn08
3bd00b88c2
support for controlnet in sample output
2023-06-01 20:48:30 +09:00
ddPn08
62d00b4520
add controlnet training
2023-06-01 20:48:25 +09:00
ykume
758a1e7f66
Revert unet config, add option to convert script
2023-05-03 16:05:15 +09:00
ykume
1cba447102
fix unet cfg is different in saving diffuser model
2023-05-03 14:06:51 +09:00
Kohya S
4627b389ff
fix device not specified in merge_lora.py
2023-04-01 09:15:57 +09:00
Kohya S
1cd07770a4
format by black
2023-04-01 09:13:47 +09:00
Kohya S
8cecc676cf
Fix device issue in load_file, reduce vram usage
2023-03-31 09:05:51 +09:00
Kohya S
6c28dfb417
Merge pull request #332 from guaneec/ddp-lowram
...
Reduce peak RAM usage
2023-03-30 21:37:37 +09:00
Kohya S
895b0b6ca7
Fix saving issue if epoch/step not in checkpoint
2023-03-27 21:22:32 +09:00
guaneec
3cdae0cbd2
Reduce peak RAM usage
2023-03-27 14:34:17 +08:00
Kohya S
b885c6f9d2
disable annoying warning in CLIP loading
2023-03-09 20:25:21 +09:00
Kohya S
ea2dfd09ef
update bucketing features
2023-02-05 21:37:46 +09:00
Kohya S
6bbb4d426e
Fix unet config in Diffusers (sample_size=64)
2023-01-29 20:43:58 +09:00
Kohya S
61ec60a893
move convert_vae to inline, restore comments
2023-01-14 21:24:09 +09:00
Hacker 17082006
dfeadf9e52
.bin file don't need to be checked
2023-01-14 15:23:46 +07:00
Hacker 17082006
b3d3f0c8ac
Not necessary to edit load_checkpoint_with_text_encoder_conversion
2023-01-14 15:07:56 +07:00
Hacker 17082006
4fe1dd6a1c
Wrong indention
2023-01-14 14:59:29 +07:00
Hacker 17082006
95ee349e2a
Edit wrong file :<
2023-01-14 14:55:57 +07:00
Kohya S
f56988b252
unify dataset and save functions
2023-01-05 08:10:22 +09:00
Kohya S
f192338874
rename another position_ids key (supports wd v1.4)
2023-01-01 09:17:16 +09:00
bmaltais
c5cca931ab
Proposed file structure rework and required file changes
2022-12-18 15:24:46 -05:00