275 commits

Author	SHA1	Message	Date
Shen-Chenhui	b78b469420	fix merge	2024-04-29 17:58:59 +08:00
Shen-Chenhui	f81d8648cf	move out losses to different file	2024-04-29 17:53:03 +08:00
zhengzangw	a2dc903863	update	2024-04-29 09:41:39 +00:00
Shen-Chenhui	4d69671663	clean up code	2024-04-29 17:18:45 +08:00
zhengzangw	1171e5b6f9	update config	2024-04-29 07:27:15 +00:00
Shen-Chenhui	98de78910d	config	2024-04-29 15:21:54 +08:00
Shen-Chenhui	7811b8e99a	add Nlayer 3D discriminator	2024-04-29 14:18:57 +08:00
Shen-Chenhui	069ffcc4e3	use 3 layer disc	2024-04-29 14:03:18 +08:00
Shen-Chenhui	8b465b04c5	add vae subfolder support	2024-04-29 11:32:39 +08:00
Shen-Chenhui	6fb4e3cd22	debug	2024-04-27 17:18:19 +08:00
Shen-Chenhui	c9785342c2	add evals	2024-04-27 16:14:11 +08:00
Shen-Chenhui	38c46ac721	fix bug	2024-04-27 15:25:03 +08:00
Shen Chenhui	b5a3334caf	fix dataset loading	2024-04-27 14:52:52 +08:00
Shen Chenhui	3e7ba9a46d	use modulelist	2024-04-27 14:02:20 +08:00
zhengzangw	478b585024	[wip] debug vae	2024-04-26 07:27:26 +00:00
zhengzangw	a4652a8aef	Merge branch 'main' into vae	2024-04-26 06:56:22 +00:00
Shen-Chenhui	576b44d98e	add lecam storage support	2024-04-26 10:34:00 +08:00
Frank Lee	367ae02e7c	updated gradio app (#74)	2024-04-24 22:30:36 +08:00
zhengzangw	44e618bbbc	a bunch of update	2024-04-24 09:23:24 +00:00
Shen-Chenhui	9683ec66c7	inference script update	2024-04-24 17:04:53 +08:00
Shen-Chenhui	0138be6e21	grad penalty expr	2024-04-24 14:52:47 +08:00
zhengzangw	9e0b314dab	update config	2024-04-24 02:33:27 +00:00
Shen-Chenhui	988bc3bb65	config lc gp	2024-04-24 10:03:58 +08:00
Zangwei Zheng	e009e89e1a	add start from scratch	2024-04-23 19:24:22 +08:00
Shen-Chenhui	48e3a84ba6	enable r1 gradient penalty	2024-04-23 17:51:50 +08:00
zhengzangw	cf95f41002	update docs	2024-04-23 09:26:10 +00:00
Shen-Chenhui	d565477680	disable non-master print	2024-04-23 14:43:02 +08:00
Shen-Chenhui	d0d7da36ea	silence lpips	2024-04-23 11:44:33 +08:00
Shen-Chenhui	2ac952bb71	8 GPU training config	2024-04-23 11:30:40 +08:00
Shen-Chenhui	486593f137	LeCam Loss working	2024-04-23 10:58:09 +08:00
Shen-Chenhui	16b0149fc4	lecam config	2024-04-23 09:59:12 +08:00
Shen-Chenhui	f4fe9b5eca	add lecam	2024-04-23 09:22:05 +08:00
Zangwei Zheng	3944e93065	update training config	2024-04-22 12:05:59 +08:00
zhengzangw	8931efccd7	update config	2024-04-22 03:43:31 +00:00
Zangwei Zheng	fea6aba655	hotfix config args	2024-04-21 19:50:51 +08:00
zhengzangw	da345719c7	use qk-norm	2024-04-21 11:40:20 +00:00
zhengzangw	2b42ca82d8	[wip] prompt with json	2024-04-21 11:07:59 +00:00
zhengzangw	6073af6444	merge video edit	2024-04-20 15:47:48 +00:00
Zheng Zangwei (Alex Zheng)	341b12f9bf	Dev/v1.0.1 (#58) ... * update (#57) * update * update datautil * add VBench prompt * update eval * update eval * update intepolation * add vbench eval * Dev/sdedit implementation (#56) * Update utils.py * update * update * update --------- Co-authored-by: YuKun Zhou <90625606+1zeryu@users.noreply.github.com>	2024-04-20 21:23:10 +08:00
zhengzangw	d262f14a24	update	2024-04-20 02:43:38 +00:00
Zangwei Zheng	74aade586a	[exp] update	2024-04-19 17:56:02 +08:00
Shen-Chenhui	96a42e08db	config	2024-04-19 17:49:50 +08:00
Zangwei Zheng	15f9d702ed	[exp] update	2024-04-19 17:43:44 +08:00
Zangwei Zheng	1877a14abc	load ema strict=False	2024-04-19 16:55:03 +08:00
Zangwei Zheng	8bf34f4418	add local_files_only	2024-04-19 16:47:26 +08:00
Shen-Chenhui	4c46972b64	debug	2024-04-19 15:59:21 +08:00
Shen-Chenhui	19187b70f5	allow specifying outputs in command line	2024-04-19 15:56:04 +08:00
Zangwei Zheng	dce2ef4a1c	[exp] new config	2024-04-19 15:21:00 +08:00
Shen-Chenhui	78caeea999	add support and config for z channel 4	2024-04-19 15:14:53 +08:00
Shen-Chenhui	b16001e5e3	debug	2024-04-19 14:28:53 +08:00