1014 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
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
Kohya S	6aa62b9b66	update readme	2023-07-03 21:06:58 +09:00
Kohya S	2febbfe4b0	add error message for old npz	2023-07-03 20:58:35 +09:00
Kohya S	ea182461d3	add min/max_timestep	2023-07-03 20:44:42 +09:00
Kohya S	5863676ccb	update readme	2023-07-02 16:49:18 +09:00
Kohya S	97611e89ca	remove debug code	2023-07-02 16:49:11 +09:00
Kohya S	64cf922841	add feature to sample images during sdxl training	2023-07-02 16:42:19 +09:00
Kohya S	227a62e4c4	fix to work with dreambooth ds without toml	2023-06-30 07:40:22 +09:00
Kohya S	38e21f5c1a	update transfomer to fix sdxl text model with bf16	2023-06-29 13:03:00 +09:00
Kohya S	d395bc0647	fix max_token_length not works for sdxl	2023-06-29 13:02:19 +09:00
Yuta Hayashibe	afce13d101	Fix a typo	2023-06-28 21:17:20 +09:00
Kohya S	8521ab7990	fix to work	2023-06-28 13:09:02 +09:00