chestnutlzj commited on
Commit
65b7aee
·
verified ·
1 Parent(s): ccd0047

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +36 -0
config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "WFVAEModel",
3
+ "_diffusers_version": "0.30.2",
4
+ "attention_type": "AttnBlock3DFix",
5
+ "base_channels": 192,
6
+ "connect_res_layer_num": 1,
7
+ "decoder_energy_flow_hidden_size": 128,
8
+ "decoder_num_resblocks": 2,
9
+ "dropout": 0.0,
10
+ "encoder_energy_flow_hidden_size": 128,
11
+ "encoder_num_resblocks": 2,
12
+ "l1_dowmsample_block": "Downsample",
13
+ "l1_downsample_wavelet": "HaarWaveletTransform2D",
14
+ "l1_upsample_block": "Upsample",
15
+ "l1_upsample_wavelet": "InverseHaarWaveletTransform2D",
16
+ "l2_dowmsample_block": "Spatial2xTime2x3DDownsample",
17
+ "l2_downsample_wavelet": "HaarWaveletTransform3D",
18
+ "l2_upsample_block": "Spatial2xTime2x3DUpsample",
19
+ "l2_upsample_wavelet": "InverseHaarWaveletTransform3D",
20
+ "latent_dim": 16,
21
+ "norm_type": "layernorm",
22
+ "scale": [
23
+ 0.18215,
24
+ 0.18215,
25
+ 0.18215,
26
+ 0.18215
27
+ ],
28
+ "shift": [
29
+ 0,
30
+ 0,
31
+ 0,
32
+ 0
33
+ ],
34
+ "t_interpolation": "trilinear",
35
+ "use_attention": true
36
+ }