LanguageBind commited on
Commit
2641787
1 Parent(s): b6092d3

Upload 2 files

Browse files
Files changed (2) hide show
  1. vae/config.json +15 -0
  2. vae/merged.ckpt +3 -0
vae/config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "WFVAEModel",
3
+ "_diffusers_version": "0.28.0",
4
+ "base_channels": 128,
5
+ "decoder_energy_flow_hidden_size": 128,
6
+ "decoder_num_resblocks": 2,
7
+ "dropout": 0.0,
8
+ "encoder_energy_flow_hidden_size": 64,
9
+ "encoder_num_resblocks": 2,
10
+ "latent_dim": 8,
11
+ "use_attention": true,
12
+ "norm_type": "layernorm",
13
+ "t_interpolation": "trilinear",
14
+ "connect_res_layer_num": 2
15
+ }
vae/merged.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cb45b11506efe8081b0371cdec6663fd73cc2ee2a3eb41767fec9db16de72b68
3
+ size 1030159594