736 Commits

Author	SHA1	Message	Date
AI-Casanova	a3c7d711e4	Min-SNR Weighting Strategy: Fixed SNR calculation to authors implementation	2023-03-23 05:43:46 +00:00
u-haru	dbadc40ec2	persistent_workersを有効にした際にキャプションが変化しなくなるバグ修正	2023-03-23 12:33:03 +09:00
u-haru	447c56bf50	typo修正、stepをglobal_stepに修正、バグ修正	2023-03-23 09:53:14 +09:00
u-haru	a9b26b73e0	implement token warmup	2023-03-23 07:37:14 +09:00
AI-Casanova	64c923230e	Min-SNR Weighting Strategy: Refactored and added to all trainers	2023-03-22 01:27:29 +00:00
AI-Casanova	795a6bd2d8	Merge branch 'kohya-ss:main' into min-SNR	2023-03-21 13:19:15 -05:00
Kohya S	7b324bcc3b	support extensions of image files with uppercases	2023-03-21 21:10:34 +09:00
Kohya S	6d9f3bc0b2	fix different reso in batch	2023-03-21 18:33:46 +09:00
Kohya S	1816ac3271	add vae_batch_size option for faster caching	2023-03-21 18:15:57 +09:00
Kohya S	cb08fa0379	fix no npz with full path	2023-03-21 15:05:25 +09:00
AI-Casanova	a265225972	Min-SNR Weighting Strategy	2023-03-20 22:51:38 +00:00
Kohya S	de95431895	support win with diffusers, fix extra args eval	2023-03-19 22:09:36 +09:00
Kohya S	48c1be34f3	Merge branch 'dev' into main	2023-03-19 21:58:41 +09:00
Kohya S	140b4fad43	remove default values from output config	2023-03-19 20:06:31 +09:00
Kohya S	1f7babd2c7	Fix lpwp to support sdv2 and clip skip	2023-03-19 11:10:17 +09:00
Kohya S	1214760cea	Merge branch 'dev' into main	2023-03-19 10:56:56 +09:00
Kohya S	64d85b2f51	fix num_processes, fix indent	2023-03-19 10:52:46 +09:00
Kohya S	ec7f9bab6c	Merge branch 'dev' into dev	2023-03-19 10:25:22 +09:00
Kohya S	83e102c691	refactor config parse, feature to output config	2023-03-19 10:11:11 +09:00
Kohya S	c3f9eb10f1	format with black	2023-03-18 18:58:12 +09:00
orenwang	370ca9e8cd	fix exception on training model in diffusers format	2023-03-13 14:32:43 +08:00
mio	e24a43ae0b	sample images with weight and no length limit	2023-03-12 16:08:31 +08:00
Linaqruf	44d4cfb453	feat: added function to load training config with .toml	2023-03-12 11:52:37 +07:00
Kohya S	618592c52b	npz check to use subset, add dadap warn close #274	2023-03-10 21:31:59 +09:00
Kohya S	e355b5e1d3	Merge pull request #269 from rvhfxb/patch-2 ... Allow to delete images after getting latents	2023-03-10 20:56:11 +09:00
Isotr0py	e3b2bb5b80	Merge branch 'dev' into dev	2023-03-10 19:04:07 +08:00
Isotr0py	7544b38635	fix multi gpu	2023-03-10 18:45:53 +08:00
Isotr0py	c4a596df9e	replace unsafe eval() with ast	2023-03-10 13:44:16 +08:00
Kohya S	458173da5e	Merge branch 'dev' into dev	2023-03-10 13:00:49 +09:00
Kohya S	51249b1ba0	support conv2d 3x3 LoRA	2023-03-09 20:56:33 +09:00
Isotr0py	ab05be11d2	fix wrong typing	2023-03-09 19:35:06 +08:00
Kohya S	b885c6f9d2	disable annoying warning in CLIP loading	2023-03-09 20:25:21 +09:00
Isotr0py	eb68892ab1	add lr_scheduler_type etc	2023-03-09 16:51:22 +08:00
rvhfxb	82aac26469	Update train_util.py	2023-03-08 22:42:41 +09:00
Kohya S	8929bf31d9	sample gen h/w to div by 8, fix in steps=epoch	2023-03-08 21:18:28 +09:00
ddPn08	87846c043f	fix for multi gpu training	2023-03-08 09:46:37 +09:00
Kohya S	225c533279	accept empty caption #258	2023-03-07 08:23:34 +09:00
Kohya S	8d5ba29363	free pipe and cache after sample gen #260	2023-03-07 08:06:36 +09:00
Kohya S	46aee85d2a	re2-fix to support python 3.8/3.9	2023-03-05 23:27:16 +09:00
Kohya S	2ae33db83f	re-fix to support python 3.8/3.9	2023-03-05 22:35:32 +09:00
Kohya S	dd39e5d944	hope to support python 3.8/3.9	2023-03-05 20:04:18 +09:00
Kohya S	5602e0e5fc	change dataset config option to dataset_config	2023-03-02 21:51:58 +09:00
Kohya S	2d2407410e	show index in caching latents	2023-03-02 21:32:02 +09:00
Kohya S	859f8361bb	minor fix in token shuffling	2023-03-02 20:31:07 +09:00
Kohya S	c3024be8bf	add help for keep_tokens	2023-03-02 20:28:42 +09:00
Kohya S	83bfb54f20	fix num_repeats not working in DB classic dataset	2023-03-02 19:01:22 +09:00
Kohya S	04af36e7e2	strip tag, fix tag frequency count	2023-03-01 22:10:15 +09:00
Kohya S	d1d7d432e9	print dataset index in making buckets	2023-03-01 21:30:12 +09:00
Kohya S	089a63c573	shuffle at debug_dataset	2023-03-01 21:12:33 +09:00
Kohya S	ed19a92bbe	fix typos	2023-03-01 21:01:10 +09:00