Commit graph

24 commits

Author SHA1 Message Date
Shen-Chenhui
7811b8e99a add Nlayer 3D discriminator 2024-04-29 14:18:57 +08:00
Shen-Chenhui
c9785342c2 add evals 2024-04-27 16:14:11 +08:00
Shen-Chenhui
48e3a84ba6 enable r1 gradient penalty 2024-04-23 17:51:50 +08:00
Shen-Chenhui
2ac952bb71 8 GPU training config 2024-04-23 11:30:40 +08:00
Shen-Chenhui
486593f137 LeCam Loss working 2024-04-23 10:58:09 +08:00
Shen-Chenhui
16b0149fc4 lecam config 2024-04-23 09:59:12 +08:00
Shen-Chenhui
41a600b61b inference config 2024-04-19 10:47:09 +08:00
Shen-Chenhui
af1b1e484d reimplement blurpool 2024-04-15 17:47:33 +08:00
Shen-Chenhui
ccb92cd1ab debug 2024-04-11 15:34:11 +08:00
Shen-Chenhui
63f3737eb0 add simple discriminator 2024-04-11 15:14:16 +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
9091419b28 save vae to model dir 2024-04-05 16:07:23 +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