Yidi
13df47516d
Remove position_ids for V2
...
The postions_ids cause errors for the newer version of transformer.
This has already been fixed in convert_ldm_clip_checkpoint_v1() but
not in v2.
The new code applies the same fix to convert_ldm_clip_checkpoint_v2().
2025-02-20 04:49:51 -05:00
Kohya S
358ca205a3
Merge branch 'dev' into dev_device_support
2024-02-12 13:01:54 +09:00
Yuta Hayashibe
5f6bf29e52
Replace print with logger if they are logs ( #905 )
...
* Add get_my_logger()
* Use logger instead of print
* Fix log level
* Removed line-breaks for readability
* Use setup_logging()
* Add rich to requirements.txt
* Make simple
* Use logger instead of print
---------
Co-authored-by: Kohya S <52813779+kohya-ss@users.noreply.github.com >
2024-02-04 18:14:34 +09:00
Disty0
a6a2b5a867
Fix IPEX support and add XPU device to device_utils
2024-01-31 17:32:37 +03:00
Aarni Koskela
6f3f701d3d
Deduplicate ipex initialization code
2024-01-19 18:07:36 +02:00
Kohya S
aab265e431
Fix an issue with saving as diffusers sd1/2 model close #1033
2024-01-04 21:43:50 +09:00
Kohya S
716bad188b
Update dependencies ref #1024
2024-01-04 19:53:25 +09:00
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