Commit graph

31 commits

Author SHA1 Message Date
zhengzangw
f77370fff6 [feat] vae micro frame 2024-05-05 15:25:17 +00:00
zhengzangw
02904b4f29 updated 2024-05-02 03:46:01 +00:00
zhengzangw
1f9c0ab007 update inference vae 2024-04-30 08:37:07 +00:00
zhengzangw
ace37cf7c7 refactor 2024-04-30 08:13:20 +00:00
zhengzangw
41e276f5ef updated 2024-04-30 06:19:40 +00:00
zhengzangw
b2f302b379 [wip] try vae 2024-04-29 17:02:22 +00:00
Shen-Chenhui
c59b499085 Merge branch 'vae-clean-new' of https://github.com/hpcaitech/Open-Sora-dev into vae-clean-new 2024-04-29 18:10:03 +08:00
Shen-Chenhui
c631358bbf add z nll loss 2024-04-29 18:09:42 +08:00
zhengzangw
9632b5bcf8 merged 2024-04-29 10:07:09 +00:00
zhengzangw
5c8139521a update 2024-04-29 10:02:42 +00:00
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
Shen-Chenhui
a4025b0ea7 magvit v2 enc dec arc 2024-04-09 17:49:01 +08:00
Shen-Chenhui
a1808be460 debug 2024-04-08 18:18:05 +08:00
Shen-Chenhui
4861e6beaf debug 2024-04-08 18:13:15 +08:00
Shen-Chenhui
9f211646f3 debug 2024-04-08 18:11:27 +08:00
Shen-Chenhui
ab38a411a8 debug 2024-04-08 18:07:50 +08:00
Shen-Chenhui
bbeacb491d debug 2024-04-08 17:50:41 +08:00
Shen-Chenhui
475762cdb2 debug 2024-04-08 16:37:26 +08:00
Shen-Chenhui
c6181ebdfd debug 2024-04-08 16:36:54 +08:00
Shen-Chenhui
90fc92ee9c debug 2024-04-08 16:14:00 +08:00
Shen-Chenhui
ab1970ca24 debug 2024-04-08 16:05:15 +08:00
Shen-Chenhui
4b27448a49 debug 2024-04-08 14:22:21 +08:00
Shen-Chenhui
1b66919091 debug 2024-04-08 10:53:44 +08:00
Shen-Chenhui
ad02d28d9c debug inference running loss 2024-04-08 10:22:07 +08:00
Shen-Chenhui
e3584b4e43 debug 2024-04-05 16:20:39 +08:00
Shen-Chenhui
bc095aa492 debug 2024-04-04 15:26:56 +08:00
Shen-Chenhui
dfdf8db491 add inference config
add readme
complete inference script
TODO: DEBUG inference
2024-04-04 14:11:25 +08:00
Shen-Chenhui
b6c39873ee finish discriminator arch
use einops.rearrange in vae model_utils
add get_latent_size in vae_3d
inference vae_3d WIP
2024-04-03 10:29:01 +08:00