1386 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Kohya S	13f49d1e4a	update readme	2024-02-22 23:50:10 +09:00
Kohya S	df7648245e	update readme	2024-02-22 23:41:46 +09:00
Kohya S	3368fb1af7	Modify nn.MHA to attn with q/k/v	2024-02-22 23:39:28 +09:00
Kohya S	417f14d245	Merge pull request #1130 from sdbds/fixbugs ... [stable-cascade]add save parser and fix lora scripts model name and hash	2024-02-22 12:30:59 +09:00
青龍聖者@bdsqlsz	86503cb945	add save parser and fix lora scripts model name and hash	2024-02-21 19:38:12 +08:00
Kohya S	d91b1d3793	update readme	2024-02-20 22:39:57 +09:00
Kohya S	70917077a6	update readme	2024-02-20 22:38:36 +09:00
Kohya S	69dbc50912	fix effnet encoder preprocess issue	2024-02-20 22:34:06 +09:00
Kohya S	985761ca43	fix to work without network module	2024-02-20 20:33:03 +09:00
Kohya S	71e03559e2	support LoRA training for Stable Cascade Stage C	2024-02-20 08:27:11 +09:00
Kohya S	806a6237fb	minor fixes	2024-02-18 21:57:16 +09:00
Kohya S	9b0e532942	add command line sample	2024-02-18 21:40:36 +09:00
Kohya S	c26f01241f	input prompt from console	2024-02-18 21:29:46 +09:00
Kohya S	ac71168939	add train_text_encoder arg	2024-02-18 21:29:10 +09:00
Kohya S	4e37d950d2	fix typos	2024-02-18 18:02:20 +09:00
Kohya S	4b5784eb44	update stable cascade stage C training #1119	2024-02-18 17:54:21 +09:00
Kohya S	856df07f49	Merge branch 'dev' into stable-cascade	2024-02-18 09:15:12 +09:00
Kohya S	d1fb480887	format by black	2024-02-18 09:13:24 +09:00
Kohya S	80ef59c115	support text encoder training in stable cascade	2024-02-18 09:12:37 +09:00
Kohya S	319bbf8057	add stage c tmp training code	2024-02-17 23:59:20 +09:00
Kohya S	fa440208b7	add inference script	2024-02-17 17:57:30 +09:00
Kohya S	75e4a951d0	update readme	2024-02-17 12:04:12 +09:00
Kohya S	42f3318e17	Merge pull request #1116 from kohya-ss/dev_device_support ... Dev device support	2024-02-17 11:58:02 +09:00
Kohya S	baa0e97ced	Merge branch 'dev' into dev_device_support	2024-02-17 11:54:07 +09:00
Kohya S	71ebcc5e25	update readme and gradual latent doc	2024-02-12 14:52:19 +09:00
Kohya S	93bed60762	fix to work --console_log_xxx options	2024-02-12 14:49:29 +09:00
Kohya S	41d32c0be4	Merge pull request #1117 from kohya-ss/gradual_latent_hires_fix ... Gradual latent hires fix	2024-02-12 14:21:27 +09:00
Kohya S	cbe9c5dc06	supprt deep shink with regional lora, add prompter module	2024-02-12 14:17:27 +09:00
Kohya S	d3745db764	add args for logging	2024-02-12 13:15:21 +09:00
Kohya S	358ca205a3	Merge branch 'dev' into dev_device_support	2024-02-12 13:01:54 +09:00
Kohya S	c748719115	fix indent	2024-02-12 12:59:45 +09:00
Kohya S	98f42d3a0b	Merge branch 'dev' into gradual_latent_hires_fix	2024-02-12 12:59:25 +09:00
Kohya S	35c6053de3	Merge pull request #1104 from kohya-ss/dev_improve_log ... replace print with logger	2024-02-12 11:33:32 +09:00
Kohya S	20ae603221	Merge branch 'dev' into gradual_latent_hires_fix	2024-02-12 11:26:36 +09:00
Kohya S	672851e805	Merge branch 'dev' into dev_improve_log	2024-02-12 11:24:33 +09:00
Kohya S	e579648ce9	fix help for highvram arg	2024-02-12 11:12:41 +09:00
Kohya S	e24d9606a2	add clean_memory_on_device and use it from training	2024-02-12 11:10:52 +09:00
Kohya S	75ecb047e2	Merge branch 'dev' into dev_device_support	2024-02-11 19:51:28 +09:00
Kohya S	f897d55781	Merge pull request #1113 from kohya-ss/dev_multi_gpu_sample_gen ... Dev multi gpu sample gen	2024-02-11 19:49:08 +09:00
Kohya S	7202596393	log to print tag frequencies	2024-02-10 09:59:12 +09:00
Kohya S	5d9e2873f6	make rich to output to stderr instead of stdout	2024-02-08 21:38:02 +09:00
Kohya S	055f02e1e1	add logging args for training scripts	2024-02-08 21:16:42 +09:00
Kohya S	9b8ea12d34	update log initialization without rich	2024-02-08 21:06:39 +09:00
Kohya S	74fe0453b2	add comment for get_preferred_device	2024-02-08 20:58:54 +09:00
Kohya S	efd3b58973	Add logging arguments and update logging setup	2024-02-04 20:44:10 +09:00
Kohya S	6279b33736	fallback to basic logging if rich is not installed	2024-02-04 18:28:54 +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
Kohya S	e793d7780d	reduce peak VRAM in sample gen	2024-02-04 17:31:01 +09:00
Kohya S	6269682c56	unificaition of gen scripts for SD and SDXL, work in progress	2024-02-03 23:33:48 +09:00
Kohya S	2f9a344297	fix typo	2024-02-03 23:26:57 +09:00