Commit graph

16 commits

Author SHA1 Message Date
zhengzangw
525e29abbe reformat and update docs 2024-06-17 15:37:23 +00: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
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
f489cb0915 [feat] accelerate search batch size 2024-05-15 08:07:49 +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
Watson
3eb17209f4 fix def get_model_numel() typing hint to Tuple[int, int] (#135) 2024-03-23 14:22:41 +08:00
pxy
8509094de2 update utils 2024-03-16 16:13:34 +08:00
Zheng Zangwei (Alex Zheng)
d851a85535 format (#69) 2024-03-15 22:16:20 +08:00
Zheng Zangwei (Alex Zheng)
0ba56f5309 first commit (#67) 2024-03-15 22:06:36 +08:00