1475 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Kohya S	0ec7166098	make crop top/left same as stabilityai's prep	2023-07-18 21:39:36 +09:00
Kohya S	3d66a234b0	enable different prompt for text encoders	2023-07-18 21:39:01 +09:00
Pam	8a073ee49f	vram leak fix	2023-07-17 17:51:26 +05:00
Kohya S	7e20c6d1a1	add convenience function to merge LoRA	2023-07-17 10:30:57 +09:00
Kohya S	1d4672d747	fix typos	2023-07-17 09:05:50 +09:00
Kohya S	39e62b948e	add lora for Diffusers	2023-07-16 19:57:21 +09:00
Kohya S	41d195715d	fix scheduler steps with gradient accumulation	2023-07-16 15:56:29 +09:00
Kohya S	3db97f8897	update readme	2023-07-16 15:14:49 +09:00
Kohya S	516f64f4d9	add caching to disk for text encoder outputs	2023-07-16 14:53:47 +09:00
Kohya S	62dd99bee5	update readme	2023-07-15 18:34:13 +09:00
Kohya S	94c151aea3	refactor caching latents (flip in same npz, etc)	2023-07-15 18:28:33 +09:00
Kohya S	81fa54837f	fix sampling in multi GPU training	2023-07-15 11:21:14 +09:00
Kohya S	9de357e373	fix tokenizer 2 is not same as open clip tokenizer	2023-07-14 12:27:19 +09:00
Kohya S	b4a3824ce4	change tokenizer from open clip to transformers	2023-07-13 20:49:26 +09:00
Kohya S	3bb80ebf20	fix sampling gen fails in lora training	2023-07-13 19:02:34 +09:00
Henrik Forstén	cdffd19f61	Cast weights to correct precision before transferring them to GPU	2023-07-13 12:45:28 +03:00
kaibioinfo	a7ce2633f3	fix typo in sdxl_train_textual_inversion ... bug appears when continue training on an existing TI	2023-07-12 15:06:20 +02:00
Kohya S	8fa5fb2816	support diffusers format for SDXL	2023-07-12 21:57:14 +09:00
Kohya S	8df948565a	remove unnecessary code	2023-07-12 21:53:02 +09:00
Kohya S	3c67e595b8	fix gradient accumulation doesn't work	2023-07-12 21:35:57 +09:00
Kohya S	814996b14f	fix NaN in sampling image	2023-07-11 23:18:35 +09:00
Kohya S	2e67d74df4	add no_half_vae option	2023-07-11 22:19:14 +09:00
ddPn08	b841dd78fe	make tracker init_kwargs configurable	2023-07-11 10:21:45 +09:00
Kohya S	68ca0ea995	Fix to show template type	2023-07-10 22:28:26 +09:00
Kohya S	f54b784d88	support textual inversion training	2023-07-10 22:04:02 +09:00
Kohya S	b6e328ea8f	don't hold latent on memory for finetuning dataset	2023-07-10 08:46:15 +09:00
Kohya S	5c80117fbd	update readme	2023-07-09 21:37:46 +09:00
Kohya S	c2ceb6de5f	fix uncond/cond order	2023-07-09 21:14:12 +09:00
Kohya S	77ec70d145	fix conditioning	2023-07-09 19:00:38 +09:00
Kohya S	a380502c01	fix pad token is not handled	2023-07-09 18:13:49 +09:00
Kohya S	0416f26a76	support multi gpu in caching text encoder outputs	2023-07-09 16:02:56 +09:00
Kohya S	3579b4570f	Merge pull request #628 from KohakuBlueleaf/full_bf16 ... Full bf16 support	2023-07-09 14:22:44 +09:00
Kohya S	256ff5b56c	Merge pull request #626 from ddPn08/sdxl ... support avif	2023-07-09 14:14:28 +09:00
Kohya S	7502f662ab	Merge branch 'sdxl' of https://github.com/kohya-ss/sd-scripts into sdxl	2023-07-09 14:12:05 +09:00
Kohaku-Blueleaf	d974959738	Update train_util.py for full_bf16 support	2023-07-09 12:47:26 +08:00
Kohaku-Blueleaf	5f348579d1	Update sdxl_train.py	2023-07-09 12:46:35 +08:00
ykume	8371a7a3aa	update readme	2023-07-09 13:38:48 +09:00
ykume	1d25703ac3	add generation script	2023-07-09 13:33:26 +09:00
ykume	fe7ede5af3	fix wrapper tokenizer not work for weighted prompt	2023-07-09 13:33:16 +09:00
ddPn08	d599394f60	support avif	2023-07-08 15:47:56 +09:00
Kohya S	66c03be45f	Fix TE key names for SD1/2 LoRA are invalid	2023-07-08 09:56:38 +09:00
Kohya S	c1d62383c6	update readme	2023-07-07 21:17:56 +09:00
Kohya S	73ab110260	Merge branch 'sdxl' of https://github.com/kohya-ss/sd-scripts into sdxl	2023-07-07 21:16:49 +09:00
Kohya S	cc3d40ca44	support sdxl in prepare scipt	2023-07-07 21:16:41 +09:00
Kohya S	288efddf2f	Update README.md	2023-07-06 07:43:30 +09:00
Kohya S	4a34e5804e	fix to work with .ckpt from comfyui	2023-07-05 21:55:43 +09:00
Kohya S	3d0375daa6	fix to work sdxl state dict without logit_scale	2023-07-05 21:45:30 +09:00
Kohya S	3060eb5baf	remove debug print	2023-07-05 21:44:46 +09:00
Kohya S	ce46aa0c3b	remove debug print	2023-07-04 21:34:18 +09:00
Kohya S	3b35547da0	fix dtype for vae	2023-07-04 21:30:37 +09:00