Correct `sample_size` of Stable Diffusion 2's unet to have correct width and height default

#6
Files changed (1) hide show
  1. unet/config.json +1 -1
unet/config.json CHANGED
@@ -35,7 +35,7 @@
35
  "num_class_embeds": null,
36
  "only_cross_attention": false,
37
  "out_channels": 4,
38
- "sample_size": 64,
39
  "up_block_types": [
40
  "UpBlock2D",
41
  "CrossAttnUpBlock2D",
 
35
  "num_class_embeds": null,
36
  "only_cross_attention": false,
37
  "out_channels": 4,
38
+ "sample_size": 96,
39
  "up_block_types": [
40
  "UpBlock2D",
41
  "CrossAttnUpBlock2D",