patrickvonplaten commited on
Commit
783d1c3
·
1 Parent(s): d88b335

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

Browse files

Since `diffusers==0.9.0` the width and height is automatically inferred from the `sample_size` attribute of your unet's config. It seems like your diffusion model has the same architecture as Stable Diffusion 2 which means that when using this model, by default an image size of 768x768 should be generated. This in turn means the unet's sample size should be 96.

In order to suppress to update your configuration on the fly and to suppress the deprecation warning added in this PR: https://github.com/huggingface/diffusers/pull/1406/files#r1035703505 it is strongly recommended to merge this PR.

Files changed (1) hide show
  1. unet/config.json +1 -1
unet/config.json CHANGED
@@ -33,7 +33,7 @@
33
  "norm_eps": 1e-05,
34
  "norm_num_groups": 32,
35
  "out_channels": 4,
36
- "sample_size": 34,
37
  "up_block_types": [
38
  "UpBlock2D",
39
  "CrossAttnUpBlock2D",
 
33
  "norm_eps": 1e-05,
34
  "norm_num_groups": 32,
35
  "out_channels": 4,
36
+ "sample_size": 96,
37
  "up_block_types": [
38
  "UpBlock2D",
39
  "CrossAttnUpBlock2D",