57 Commits

Author	SHA1	Message	Date
rockerBOO	c149cf283b	Add parser args for other trainers.	2025-08-03 00:58:25 -04:00
Kohya S.	93a4efabb5	Merge branch 'sd3' into resize-interpolation	2025-03-30 19:30:56 +09:00
Kohya S	ba5251168a	fix: save tensors as is dtype, add save_precision option	2025-03-01 10:31:39 +09:00
Kohya S	734333d0c9	feat: enhance merging logic for safetensors models to handle key prefixes correctly	2025-02-28 23:52:29 +09:00
Kohya S	5228db1548	feat: add script to merge multiple safetensors files into a single file for SD3	2025-02-26 20:50:58 +09:00
rockerBOO	7f2747176b	Use resize_image where resizing is required	2025-02-19 14:20:40 -05:00
rockerBOO	9fde0d7972	Handle tuple return from generate_dataset_group_by_blueprint	2025-01-08 18:38:20 -05:00
Kohya S	2cb7a6db02	feat: add block swap for FLUX.1/SD3 LoRA training	2024-11-12 21:39:13 +09:00
kohya-ss	bfc3a65acd	fix to work cache latents/text encoder outputs	2024-10-13 19:08:16 +09:00
kohya-ss	74228c9953	update cache_latents/text_encoder_outputs	2024-10-13 16:27:22 +09:00
Kohya S	83e3048cb0	load Diffusers format, check schnell/dev	2024-10-06 21:32:21 +09:00
Kohya S	9f44ef1330	add diffusers to FLUX.1 conversion script	2024-09-15 13:52:23 +09:00
Kohya S.	16bb5699ac	Merge pull request #1426 from sdbds/resize ... Replacing CV2 resize to Pil resize	2024-09-07 10:22:52 +09:00
sdbds	9ca7a5b6cc	instead cv2 LANCZOS4 resize to pil resize	2024-07-20 21:59:11 +08: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	da6fea3d97	simplify and update alpha mask to work with various cases	2024-05-19 21:26:18 +09:00
Kohya S	f4132018c5	fix to work with cpu_count() == 1 closes #1134	2024-02-24 19:25:31 +09:00
Kohya S	baa0e97ced	Merge branch 'dev' into dev_device_support	2024-02-17 11:54:07 +09:00
Kohya S	358ca205a3	Merge branch 'dev' into dev_device_support	2024-02-12 13:01:54 +09:00
Kohya S	98f42d3a0b	Merge branch 'dev' into gradual_latent_hires_fix	2024-02-12 12:59:25 +09:00
Yuta Hayashibe	5f6bf29e52	Replace print with logger if they are logs (#905) ... * Add get_my_logger() * Use logger instead of print * Fix log level * Removed line-breaks for readability * Use setup_logging() * Add rich to requirements.txt * Make simple * Use logger instead of print --------- Co-authored-by: Kohya S <52813779+kohya-ss@users.noreply.github.com>	2024-02-04 18:14:34 +09:00
Disty0	a6a2b5a867	Fix IPEX support and add XPU device to device_utils	2024-01-31 17:32:37 +03:00
Aarni Koskela	478156b4f7	Refactor device determination to function; add MPS fallback	2024-01-23 14:29:03 +02:00
Kohya S	da9b34fa26	Merge branch 'dev' into gradual_latent_hires_fix	2024-01-04 19:53:46 +09:00
Kohya S	3b5c1a1d4b	Fix issue with tools/convert_diffusers20_original_sd.py	2023-12-24 21:45:51 +09:00
Disty0	7080e1a11c	Fix convert_diffusers20_original_sd.py and add metadata & variant options	2023-12-22 22:40:03 +03:00
Kohya S	07ef03d340	fix controlnet to work with gradual latent	2023-12-12 08:03:27 +09:00
Yuta Hayashibe	27f9b6ffeb	updated typos to v1.16.15 and fix typos	2023-10-01 21:51:24 +09:00
Kohya S	0ecfd91a20	fix VAE becomes last one	2023-09-13 17:59:14 +09:00
Kohya S	7ea38f90d7	add merge script	2023-08-07 23:40:49 +09:00
Kohya S	c142dadb46	support sai model spec	2023-08-06 21:50:05 +09:00
Kohya S	cf6832896f	fix ControlNet with regional LoRA	2023-08-03 21:48:11 +09:00
Kohya S	0eacadfa99	fix ControlNet not working	2023-07-30 14:09:43 +09:00
Kohya S	2a4ae88f18	format by black	2023-07-30 14:03:54 +09:00
Kohya S	acf16c063a	make to work with PyTorch 1.12	2023-07-20 21:41:16 +09:00
Kohya S	516f64f4d9	add caching to disk for text encoder outputs	2023-07-16 14:53:47 +09:00
Kohya S	94c151aea3	refactor caching latents (flip in same npz, etc)	2023-07-15 18:28:33 +09:00
Kohya S	ed5bfda372	Fix controlnet input to rgb from bgr	2023-05-11 21:12:06 +09:00
ykume	758a1e7f66	Revert unet config, add option to convert script	2023-05-03 16:05:15 +09:00
ykume	1cba447102	fix unet cfg is different in saving diffuser model	2023-05-03 14:06:51 +09:00
Kohya S	a85fcfe05f	fix latent upscale not working if bs>1	2023-04-25 08:10:21 +09:00
Kohya S	884e6bff5d	fix face_crop_aug not working on finetune method, prepare upscaler	2023-04-22 10:41:36 +09:00
Kohya S	47d61e2c02	format by black	2023-04-17 22:00:26 +09:00
Kohya S	8f6fc8daa1	add ja comment, restore arg for backward compat	2023-04-17 21:58:55 +09:00
Linaqruf	7f8e05ccad	feat: add --save_precision args	2023-04-12 05:43:19 +07:00
Kohya S	4f92b6266c	fix do not starting script	2023-03-21 21:29:10 +09:00
Robert Smieja	eb66e5ebac	Extract parser setup to helper function ... - Allows users who `import` the scripts to examine the parser programmatically	2023-03-20 00:06:47 -04:00
Kohya S	014fd3d037	support original controlnet	2023-02-20 12:54:44 +09:00
Kohya S	cebee02698	Official weights to LoRA	2023-02-13 23:38:38 +09:00