Commit graph

99 commits

Author SHA1 Message Date
Shen-Chenhui
dbe1c4cf2f debug 2024-04-11 16:33:22 +08:00
Shen-Chenhui
4741d952d8 debug 2024-04-11 16:25:06 +08:00
Shen-Chenhui
158bbfa2f0 debug 2024-04-11 16:00:42 +08:00
Shen-Chenhui
38c1572e66 debug 2024-04-11 15:51:23 +08:00
Shen-Chenhui
c0632b3fb3 debug 2024-04-11 15:50:11 +08:00
Shen-Chenhui
772a52aafa debug 2024-04-11 15:46:27 +08:00
Shen-Chenhui
a769ca6d86 debug 2024-04-11 15:42:37 +08:00
Shen-Chenhui
1dd6170c89 debug 2024-04-11 15:39:51 +08:00
Shen-Chenhui
f8a1cc34b2 debug 2024-04-11 15:38:40 +08:00
Shen-Chenhui
f6c79a75c2 debug 2024-04-11 15:37:22 +08:00
Shen-Chenhui
4cb946c547 debug 2024-04-11 15:35:55 +08:00
Shen-Chenhui
ccb92cd1ab debug 2024-04-11 15:34:11 +08:00
Shen-Chenhui
9cb8e42622 add kl loss 2024-04-11 15:25:50 +08:00
Shen-Chenhui
63f3737eb0 add simple discriminator 2024-04-11 15:14:16 +08:00
Shen-Chenhui
dc63a2b4e5 debug 2024-04-11 11:28:05 +08:00
Shen-Chenhui
ca1d5863c3 add perceptual loss 2024-04-11 11:23:40 +08:00
Shen-Chenhui
e5f881ca1a debug 2024-04-11 10:53:14 +08:00
Shen-Chenhui
ee58ec16b4 add image encoding support 2024-04-11 10:50:23 +08:00
Shen-Chenhui
fef0bf875b debug 2024-04-09 18:28:19 +08:00
Shen-Chenhui
84c6918e2e debug 2024-04-09 18:26:21 +08:00
Shen-Chenhui
5064626844 debug padding 2024-04-09 18:24:39 +08:00
Shen-Chenhui
1bbef18f15 strides debug 2024-04-09 18:14:25 +08:00
Shen-Chenhui
d8666a73c4 register module 2024-04-09 18:04:08 +08:00
Shen-Chenhui
0fdc6e9935 config 2024-04-09 17:56:42 +08:00
Shen-Chenhui
a4025b0ea7 magvit v2 enc dec arc 2024-04-09 17:49:01 +08:00
Shen-Chenhui
ae6bffb8c5 restore train vae script 2024-04-09 09:34:59 +08:00
Shen-Chenhui
47b51ba07d debug 2024-04-08 16:53:18 +08:00
Shen-Chenhui
c6181ebdfd debug 2024-04-08 16:36:54 +08:00
Shen-Chenhui
fa0ca3983e debug inference code 2024-04-08 15:40:14 +08:00
Shen-Chenhui
4b27448a49 debug 2024-04-08 14:22:21 +08:00
Shen-Chenhui
ad02d28d9c debug inference running loss 2024-04-08 10:22:07 +08:00
Shen-Chenhui
88be217a13 debug 2024-04-05 16:25:32 +08:00
Shen-Chenhui
e3584b4e43 debug 2024-04-05 16:20:39 +08:00
Shen-Chenhui
9091419b28 save vae to model dir 2024-04-05 16:07:23 +08:00
Shen-Chenhui
44b8142b4f add wrapper for pretrained loading 2024-04-05 16:01:24 +08:00
Shen-Chenhui
c7506e2461 adding pretrained vae 2024-04-04 15:54:55 +08:00
Shen-Chenhui
a656421c83 debug from_pretrained in progress 2024-04-04 15:41:15 +08:00
Shen-Chenhui
d08d52ea03 config add is_vae 2024-04-04 15:23:57 +08:00
Shen-Chenhui
60eaffed4a debug 2024-04-04 15:11:42 +08:00
Shen-Chenhui
4250cfa9a2 debug 2024-04-04 15:10:11 +08:00
Shen-Chenhui
e9776584e4 allow vae not to have prompt file 2024-04-04 15:06:36 +08:00
Shen-Chenhui
9dc9afb254 add data path to inference config 2024-04-04 14:39:04 +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
Shen-Chenhui
aa733aa1d7 debug 2024-04-02 11:21:01 +08:00
Shen-Chenhui
562a966a77 debug 2024-04-01 16:14:11 +08:00
Shen-Chenhui
6bed1bdd0b working perceptual loss 2024-04-01 16:11:21 +08:00
Shen-Chenhui
6f460c1d05 debug 2024-04-01 16:06:47 +08:00
Shen-Chenhui
036b427b00 debug 2024-04-01 15:38:29 +08:00
Shen-Chenhui
996e6fb180 debug 2024-04-01 15:33:39 +08:00