sample size again
Browse filesactually 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
- 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":
|
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",
|