31 Commits

Author	SHA1	Message	Date
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