ehristoforu commited on
Commit
0f18339
1 Parent(s): d686267

Update unet/config.json

Browse files
Files changed (1) hide show
  1. unet/config.json +3 -2
unet/config.json CHANGED
@@ -18,8 +18,8 @@
18
  "center_input_sample": false,
19
  "class_embed_type": null,
20
  "class_embeddings_concat": false,
21
- "conv_in_kernel": 4,
22
- "conv_out_kernel": 4,
23
  "cross_attention_dim": 2048,
24
  "cross_attention_norm": null,
25
  "down_block_types": [
@@ -61,6 +61,7 @@
61
  10
62
  ],
63
  "up_block_types": [
 
64
  "CrossAttnUpBlock2D",
65
  "CrossAttnUpBlock2D",
66
  "UpBlock2D"
 
18
  "center_input_sample": false,
19
  "class_embed_type": null,
20
  "class_embeddings_concat": false,
21
+ "conv_in_kernel": 3,
22
+ "conv_out_kernel": 3,
23
  "cross_attention_dim": 2048,
24
  "cross_attention_norm": null,
25
  "down_block_types": [
 
61
  10
62
  ],
63
  "up_block_types": [
64
+ "CrossAttnUpBlock2D",
65
  "CrossAttnUpBlock2D",
66
  "CrossAttnUpBlock2D",
67
  "UpBlock2D"