103 commits

Author	SHA1	Message	Date
Frank Lee	be61f44a29	[gradio] udpated inference logic (#138) ... * [gradio] udpated inference logic * polish	2024-06-15 22:43:33 +08:00
zhengzangw	b70b749e55	[fix] loop support for score and refinement	2024-06-14 17:18:29 +00:00
Frank Lee	41e6e9492e	[gradio] supported random prompt and watermark (#136)	2024-06-14 17:57:44 +08:00
zhengzangw	c5e0475749	[feat] support for random sample	2024-06-14 08:50:40 +00:00
zhengzangw	c93e846017	[feat] fps control	2024-06-14 07:44:35 +00:00
zhengzangw	992aae7d8f	[feat] prompt refine	2024-06-14 07:37:00 +00:00
Frank Lee	a887f54711	added sp for stdit3 (#131)	2024-06-14 11:10:43 +08:00
Frank Lee	9dde960b95	added gradio (#133) ... * added gradio * added gradio	2024-06-13 16:56:30 +08:00
zhengzangw	a8a34d56b9	[fix] inference with score	2024-06-13 02:37:45 +00:00
zhengzangw	51cdd377cc	reformat	2024-06-12 09:24:53 +00:00
zhengzangw	a271faa131	[fix] condition sampling	2024-06-11 07:49:00 +00:00
Shen-Chenhui	bd182d93dd	format	2024-06-10 06:02:36 +00:00
zhengzangw	6094f5fe92	[feat] add score for inference	2024-06-09 12:01:26 +00:00
zhengzangw	ce601066c5	[stage2] update config	2024-06-05 15:49:51 +00:00
Frank Lee	ea96cdb5e4	added warmup lr scheduler (#121)	2024-06-05 10:12:52 +08:00
Zheng Zangwei (Alex Zheng)	9370fbe02d	update timer (#115)	2024-05-29 14:46:14 +08:00
Frank Lee	2e1e26a2e4	added timer for benchmarking (#114)	2024-05-29 14:43:15 +08:00
zhengzangw	5b9e753039	[feat] unify dataloader api	2024-05-20 08:40:45 +00:00
zhengzangw	5aad90acaf	[feat] extract feature	2024-05-17 09:53:48 +00:00
zhengzangw	a016325a8e	[feat] enable freezing y_embedder	2024-05-17 06:40:44 +00:00
zhengzangw	9fd4554f55	[fix] vae dtype	2024-05-16 08:50:24 +00:00
zhengzangw	0592723022	[feat] update bs search and loss eval	2024-05-15 13:05:53 +00:00
zhengzangw	f489cb0915	[feat] accelerate search batch size	2024-05-15 08:07:49 +00:00
zhengzangw	a36f80f847	[feat] notebook for inference	2024-05-14 07:24:56 +00:00
zhengzangw	e1e379d898	[feat] update inference config	2024-05-14 05:40:17 +00:00
zhengzangw	3a01cb440d	[feat] inference for rf	2024-05-13 11:39:16 +00:00
zhengzangw	6194da6504	[fix] update inference and datautil	2024-05-13 06:33:12 +00:00
zhengzangw	74dfaa879c	[fix] ema ckpt save	2024-05-12 05:15:06 +00:00
zhengzangw	2a55d672a7	[feat] update logging and bucket	2024-05-09 08:07:56 +00:00
zhengzangw	e9db7b3af1	update training script	2024-05-09 05:53:19 +00:00
zhengzangw	31228fe9c8	[refactor] fix train warnings	2024-05-09 03:10:26 +00:00
zhengzangw	322e566aed	update	2024-05-09 03:10:26 +00:00
Zangwei Zheng	eb5ff82fc5	quick update	2024-05-09 11:09:52 +08:00
Shen Chenhui	dc7b7fd64a	fix bug (#95) ... Co-authored-by: Shen-Chenhui <shen_chenhui@u.nus.edu>	2024-05-08 17:39:21 +08:00
Zheng Zangwei (Alex Zheng)	f047d5786a	[refactor] clean train.py code (#94)	2024-05-08 16:07:57 +08:00
Frank Lee	fe19568a49	fixed race condition in exp workspace (#91)	2024-05-07 17:17:43 +08:00
Zangwei Zheng	fea2541c70	[docs] update readme	2024-05-07 16:27:21 +08:00
Zangwei Zheng	10657621ef	[fix] random mask may have nothing to learn	2024-05-06 17:34:08 +08:00
Zheng Zangwei (Alex Zheng)	f941fa95dc	Merge branch 'dev/v1.2' into vae-clean-new	2024-05-04 20:26:42 +08:00
zhengzangw	4543bc217c	[feat] image generation	2024-05-04 10:10:54 +00:00
zhengzangw	f7a4fcfc3f	updated	2024-04-30 09:46:27 +00:00
zhengzangw	ace37cf7c7	refactor	2024-04-30 08:13:20 +00:00
zhengzangw	41e276f5ef	updated	2024-04-30 06:19:40 +00:00
周钰坤	59591d8afc	update	2024-04-30 13:36:41 +08:00
zhengzangw	1171e5b6f9	update config	2024-04-29 07:27:15 +00:00
zhengzangw	f3dcb4d6fb	complete 3b backbone	2024-04-29 06:00:14 +00:00
zhengzangw	d6a16a858e	[wip] multiple resolution	2024-04-27 11:43:15 +00:00
Shen-Chenhui	6fb4e3cd22	debug	2024-04-27 17:18:19 +08:00
YuKun Zhou	fddd08243b	Update ckpt_utils.py (#79) ... * Update ckpt_utils.py * update * Update 1x256x256.py * update * update * Update blocks.py * update * Update 1x2048x2048.py	2024-04-27 15:43:50 +08:00
Shen-Chenhui	38c46ac721	fix bug	2024-04-27 15:25:03 +08:00