TheLastBen radames commited on
Commit
da8e6c4
1 Parent(s): 68b1776

fix sample for API inference (#4)

Browse files

- fix sample for API inference (9e1198f58c4a9dc5a52e0ff3551949471e89e1d8)


Co-authored-by: Radamés Ajna <[email protected]>

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",