fix sample for API inference

#4
by radames - opened
Files changed (1) hide show
  1. vae/config.json +1 -1
vae/config.json CHANGED
@@ -19,7 +19,7 @@
19
  "layers_per_block": 2,
20
  "norm_num_groups": 32,
21
  "out_channels": 3,
22
- "sample_size": 512,
23
  "scaling_factor": 0.18215,
24
  "up_block_types": [
25
  "UpDecoderBlock2D",
 
19
  "layers_per_block": 2,
20
  "norm_num_groups": 32,
21
  "out_channels": 3,
22
+ "sample_size": 768,
23
  "scaling_factor": 0.18215,
24
  "up_block_types": [
25
  "UpDecoderBlock2D",