benjamin-paine commited on
Commit
64ab6fc
·
verified ·
1 Parent(s): 222d732

Update vae/config.json

Browse files
Files changed (1) hide show
  1. vae/config.json +1 -5
vae/config.json CHANGED
@@ -16,23 +16,19 @@
16
  "DownEncoderBlock3D"
17
  ],
18
  "extra_cond_dim": null,
19
- "force_upcast": true,
20
  "freeze_encoder": false,
21
  "gradient_checkpoint": false,
22
  "grouping": false,
23
  "in_channels": 3,
24
  "inflation_mode": "pad",
25
  "latent_channels": 16,
26
- "latents_mean": null,
27
- "latents_std": null,
28
  "layers_per_block": 2,
29
  "memory_limit": null,
30
- "mid_block_add_attention": true,
31
  "norm_num_groups": 32,
32
  "out_channels": 3,
33
  "sample_size": 32,
34
  "scaling_factor": 0.9152,
35
- "shift_factor": null,
36
  "slicing": null,
37
  "slicing_sample_min_size": 4,
38
  "slicing_up_num": 0,
 
16
  "DownEncoderBlock3D"
17
  ],
18
  "extra_cond_dim": null,
19
+ "force_upcast": false,
20
  "freeze_encoder": false,
21
  "gradient_checkpoint": false,
22
  "grouping": false,
23
  "in_channels": 3,
24
  "inflation_mode": "pad",
25
  "latent_channels": 16,
 
 
26
  "layers_per_block": 2,
27
  "memory_limit": null,
 
28
  "norm_num_groups": 32,
29
  "out_channels": 3,
30
  "sample_size": 32,
31
  "scaling_factor": 0.9152,
 
32
  "slicing": null,
33
  "slicing_sample_min_size": 4,
34
  "slicing_up_num": 0,