1681 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Kohya S	35b6cb0cd1	update for torchvision	2024-08-15 22:07:35 +09:00
Kohya S	8aaa1967bd	fix encoding latents closes #1456	2024-08-15 22:07:23 +09:00
Kohya S.	e2d822cad7	Merge pull request #1452 from fireicewolf/sd3-devel ... Fix AttributeError: 'T5EncoderModel' object has no attribute 'text_model', while loading T5 model in GPU.	2024-08-15 21:12:19 +09:00
Kohya S	7db4222119	add sample image generation during training	2024-08-14 22:15:26 +09:00
DukeG	9760d097b0	Fix AttributeError: 'T5EncoderModel' object has no attribute 'text_model' ... While loading T5 model in GPU.	2024-08-14 19:58:54 +08:00
Kohya S	56d7651f08	add experimental split mode for FLUX	2024-08-13 22:28:39 +09:00
kohya-ss	9711c96f96	update README	2024-08-13 21:03:17 +09:00
kohya-ss	f5ce754bc2	Merge branch 'dev' into sd3	2024-08-13 21:00:44 +09:00
kohya-ss	4cf42cc5d4	Merge branch 'sd3' of https://github.com/kohya-ss/sd-scripts into sd3	2024-08-13 21:00:21 +09:00
kohya-ss	0415d200f5	update dependencies closes #1450	2024-08-13 21:00:16 +09:00
Kohya S	a7d5dabde3	Update readme	2024-08-12 17:09:19 +09:00
kohya-ss	4af36f9632	update to work interactive mode	2024-08-12 13:24:10 +09:00
Kohya S	9e09a69df1	update README	2024-08-12 08:19:45 +09:00
Kohya S	74f91c2ff7	correct option name closes #1446	2024-08-11 21:54:10 +09:00
Kohya S	d25ae361d0	fix apply_t5_attn_mask to work	2024-08-11 19:07:07 +09:00
Kohya S	82314ac2e7	update readme for ai toolkit settings	2024-08-11 11:14:08 +09:00
Kohya S	8a0f12dde8	update FLUX LoRA training	2024-08-10 23:42:05 +09:00
Kohya S	358f13f2c9	fix alpha is ignored	2024-08-10 14:03:59 +09:00
Kohya S	808d2d1f48	fix typos	2024-08-09 23:02:51 +09:00
Kohya S	36b2e6fc28	add FLUX.1 LoRA training	2024-08-09 22:56:48 +09:00
Kohya S	da4d0fe016	support attn mask for l+g/t5	2024-08-05 20:51:34 +09:00
Kohya S	231df197dd	Fix npz path for verification	2024-08-05 20:26:30 +09:00
Kohya S	002d75179a	sample images for training	2024-07-29 23:18:34 +09:00
Kohya S	1a977e847a	fix typos	2024-07-27 13:51:50 +09:00
Kohya S	41dee60383	Refactor caching mechanism for latents and text encoder outputs, etc.	2024-07-27 13:50:05 +09:00
Kohya S	082f13658b	reduce peak GPU memory usage before training	2024-07-12 21:28:01 +09:00
Kohya S	b8896aad40	update README	2024-07-11 08:01:23 +09:00
Kohya S	6f0e235f2c	Fix shift value in SD3 inference.	2024-07-11 08:00:45 +09:00
Kohya S	3d402927ef	WIP: update new latents caching	2024-07-09 23:15:38 +09:00
Kohya S	9dc7997803	fix typo	2024-07-09 20:37:00 +09:00
Kohya S	3ea4fce5e0	load models one by one	2024-07-08 22:04:43 +09:00
Kohya S	c9de7c4e9a	WIP: new latents caching	2024-07-08 19:48:28 +09:00
Kohya S	50e3d62474	fix to work T5XXL with fp16	2024-07-08 19:46:23 +09:00
Kohya S	ea18d5ba6d	Fix to work full_bf16 and full_fp16.	2024-06-29 17:45:50 +09:00
Kohya S	19086465e8	Fix fp16 mixed precision, model is in bf16 without full_bf16	2024-06-29 17:21:25 +09:00
Kohya S	66cf435479	re-fix assertion ref #1389	2024-06-27 13:14:09 +09:00
Kohya S	381598c8bb	fix resolution in metadata for sd3	2024-06-26 21:15:02 +09:00
Kohya S	828a581e29	fix assertion for experimental impl ref #1389	2024-06-26 20:43:31 +09:00
Kohya S	8f2ba27869	support text_encoder_batch_size for caching	2024-06-26 20:36:22 +09:00
Kohya S	0b3e4f7ab6	show file name if error in load_image ref #1385	2024-06-25 20:03:09 +09:00
Kohya S	4802e4aaec	workaround for long caption ref #1382	2024-06-24 23:13:14 +09:00
Kohya S	0fe4eafac9	fix to use zero for initial latent	2024-06-24 23:12:48 +09:00
Kohya S	d53ea22b2a	sd3 training	2024-06-23 23:38:20 +09:00
Kohya S	a518e3c819	Merge branch 'dev' into sd3	2024-06-23 14:12:07 +09:00
Kohya S	9dd1ee458c	Merge branch 'main' into dev	2024-06-23 14:11:51 +09:00
Kohya S	25f961bc77	fix to work cache_latents/text_encoder_outputs	2024-06-23 13:24:30 +09:00
Kohya S	e5268286bf	add sd3 models and inference script	2024-06-15 22:20:24 +09:00
Kohya S	56bb81c9e6	add grad_hook after restore state closes #1344	2024-06-12 21:39:35 +09:00
Kohya S	22413a5247	Merge pull request #1359 from kohya-ss/train_resume_step ... Train resume step	2024-06-11 19:52:03 +09:00
Kohya S	18d7597b0b	update README	2024-06-11 19:51:30 +09:00