Commit Graph

21 Commits (b9d5db8a423a4e321546eb609bfba0c572306d24)

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