KevinNg99 commited on
Commit
0ebd534
·
1 Parent(s): 97cdeee

Add VAE checkpoint

Browse files
vae/vae_2_1/config.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "HYVAE2D",
3
+ "block_out_channels": [
4
+ 128,
5
+ 256,
6
+ 512,
7
+ 512,
8
+ 1024,
9
+ 1024
10
+ ],
11
+ "in_channels": 3,
12
+ "out_channels": 3,
13
+ "latent_channels": 64,
14
+ "layers_per_block": 2,
15
+ "ffactor_spatial": 32,
16
+ "sample_size": 384,
17
+ "sample_tsize": 96,
18
+ "scaling_factor": 0.75289,
19
+ "downsample_match_channel": true,
20
+ "upsample_match_channel": true
21
+ }
vae/vae_2_1/pytorch_model.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:85a0aec2ffb92ba1bba363c6e9c673e1b52997808a4a496e2b127b1e62733e90
3
+ size 1622406234