Commit graph

77 commits

Author SHA1 Message Date
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
zhengzangw
a4652a8aef Merge branch 'main' into vae 2024-04-26 06:56:22 +00:00
Zangwei Zheng
e009e89e1a add start from scratch 2024-04-23 19:24:22 +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
2b42ca82d8 [wip] prompt with json 2024-04-21 11:07:59 +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
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
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
Shen-Chenhui
b16001e5e3 debug 2024-04-19 14:28:53 +08:00
Shen-Chenhui
89a58b8e15 debug 2024-04-19 14:23:57 +08:00
Shen-Chenhui
1c3e910fc6 allow loading of discriminator in inference 2024-04-19 14:19:29 +08:00
Zheng Zangwei (Alex Zheng)
d2a782efac Dev/fps (#55)
* support fps

* update fps
2024-04-19 13:18:52 +08:00
Zangwei Zheng
0e6c15d50b Merge branch 'main' into dev/v1.0.1 2024-04-19 12:48:22 +08:00
Zheng Zangwei (Alex Zheng)
069ea0d687 Feat/fast bucket (#54)
* [wip] bucket

* [bug] not parallel

* update eval

* update sample.sh

* accelerate bucket build with pandarallel
2024-04-19 11:42:02 +08:00
7sunarni
d90c049a2f feat(ckpt_utils): support HF_MIRROR (#264)
Signed-off-by: 7sunarni <710720732@qq.com>
2024-04-18 23:31:40 +08:00
zhengzangw
2539a610d5 update mask strategy 2024-04-18 08:13:53 +00:00
Zheng Zangwei (Alex Zheng)
1f99b9ba4b complete eval pipeline (#53) 2024-04-18 15:49:14 +08:00
Zangwei Zheng
ee2e4083e4 [feat] update eval 2024-04-17 17:35:44 +08:00
Zangwei Zheng
8cb6d2f0bd [feat] update eval 2024-04-17 17:05:08 +08:00
Zheng Zangwei (Alex Zheng)
a72e59610f update extending ynull (#51) 2024-04-16 16:45:06 +08:00
Zangwei Zheng
50ad9a1ca5 small update 2024-04-15 17:36:20 +08:00
Zangwei Zheng
da1038ca5c a bunch of update 2024-04-11 11:48:06 +08:00