radames commited on
Commit
9f47f72
1 Parent(s): 444d448

sample size again

Browse files

actually if we want default to be 128

```width = width or self.unet.config.sample_size * 2 ** (len(self.vae.config.block_out_channels) - 1) ```

16 * 2**(4- 1) = 128

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