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 |
|