anton-l HF staff commited on
Commit
895564f
1 Parent(s): 7cb8cf8

fix sample size

Browse files
Files changed (1) hide show
  1. vae/config.json +1 -1
vae/config.json CHANGED
@@ -15,7 +15,7 @@
15
  "layers_per_block": 1,
16
  "norm_num_groups": 32,
17
  "out_channels": 3,
18
- "sample_size": 32,
19
  "up_block_types": [
20
  "UpDecoderBlock2D",
21
  "UpDecoderBlock2D"
 
15
  "layers_per_block": 1,
16
  "norm_num_groups": 32,
17
  "out_channels": 3,
18
+ "sample_size": 128,
19
  "up_block_types": [
20
  "UpDecoderBlock2D",
21
  "UpDecoderBlock2D"