72 Commits

Author	SHA1	Message	Date
Kohya S	6889ee2b85	add warning for bucket_reso_steps with SDXL	2023-08-11 19:02:36 +09:00
Kohya S	92f41f1051	update sdxl ver in lora metadata from v0-9 to v1-0	2023-08-06 22:06:48 +09:00
Kohya S	0636399c8c	add adding v-pred like loss for noise pred	2023-07-31 08:23:28 +09:00
Kohya S	b1e44e96bc	fix to show batch size for each dataset refs #637	2023-07-23 15:39:56 +09:00
Kohya S	73a08c0be0	Merge pull request #630 from ddPn08/sdxl ... make tracker init_kwargs configurable	2023-07-20 22:05:55 +09:00
Kohya S	acf16c063a	make to work with PyTorch 1.12	2023-07-20 21:41:16 +09:00
Kohya S	225e871819	enable full bf16 trainint in train_network	2023-07-19 08:41:42 +09:00
Kohya S	6d2d8dfd2f	add zero_terminal_snr option	2023-07-18 23:17:23 +09:00
Kohya S	516f64f4d9	add caching to disk for text encoder outputs	2023-07-16 14:53:47 +09:00
Kohya S	8fa5fb2816	support diffusers format for SDXL	2023-07-12 21:57:14 +09:00
Kohya S	3c67e595b8	fix gradient accumulation doesn't work	2023-07-12 21:35:57 +09:00
ddPn08	b841dd78fe	make tracker init_kwargs configurable	2023-07-11 10:21:45 +09:00
Kohya S	0416f26a76	support multi gpu in caching text encoder outputs	2023-07-09 16:02:56 +09:00
Kohaku-Blueleaf	5f348579d1	Update sdxl_train.py	2023-07-09 12:46:35 +08:00
Kohya S	3d0375daa6	fix to work sdxl state dict without logit_scale	2023-07-05 21:45:30 +09:00
Kohya S	ea182461d3	add min/max_timestep	2023-07-03 20:44:42 +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	d395bc0647	fix max_token_length not works for sdxl	2023-06-29 13:02:19 +09:00
Kohya S	a751dc25d6	use CLIPTextModelWithProjection	2023-06-27 20:48:06 +09:00
Kohya S	2c461e4ad3	Add no_half_vae for SDXL training, add nan check	2023-06-26 20:38:09 +09:00
Kohya S	747af145ed	add sdxl fine-tuning and LoRA	2023-06-26 08:07:24 +09:00