nousr commited on
Commit
246f7a1
·
1 Parent(s): af9fcbc

Upload decoder/decoder_config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. decoder/decoder_config.json +3 -3
decoder/decoder_config.json CHANGED
@@ -4,18 +4,18 @@
4
  "unets": [
5
  {
6
  "dim": 320,
7
- "cond_dim": 512,
8
  "image_embed_dim": 1024,
9
  "text_embed_dim": 1024,
10
  "cond_on_text_encodings": true,
11
  "channels": 3,
12
  "dim_mults": [1, 2, 3, 4],
13
  "num_resnet_blocks": 4,
14
- "attn_heads": 16,
15
  "attn_dim_head": 64,
16
  "sparse_attn": true,
17
  "memory_efficient": true,
18
- "scale_skip_connection":true,
19
  "resnet_weight_standardization": true,
20
  "self_attn": [false, true, true, true]
21
  }
 
4
  "unets": [
5
  {
6
  "dim": 320,
7
+ "cond_dim": 768,
8
  "image_embed_dim": 1024,
9
  "text_embed_dim": 1024,
10
  "cond_on_text_encodings": true,
11
  "channels": 3,
12
  "dim_mults": [1, 2, 3, 4],
13
  "num_resnet_blocks": 4,
14
+ "attn_heads": 8,
15
  "attn_dim_head": 64,
16
  "sparse_attn": true,
17
  "memory_efficient": true,
18
+ "scale_skip_connection": true,
19
  "resnet_weight_standardization": true,
20
  "self_attn": [false, true, true, true]
21
  }