Veldrovive commited on
Commit
351a97d
1 Parent(s): 43e8866

Upload decoder/2.4B_fp16/decoder_config.json with huggingface_hub

Browse files
decoder/2.4B_fp16/decoder_config.json CHANGED
@@ -4,7 +4,7 @@
4
  "decoder": {
5
  "unets": [
6
  {
7
- "dim": 768,
8
  "cond_dim": 512,
9
  "image_embed_dim": 768,
10
  "text_embed_dim": 768,
@@ -13,7 +13,7 @@
13
  "dim_mults": [1, 2, 3, 4],
14
  "self_attn": [false, true, true, true],
15
  "num_resnet_blocks": 4,
16
- "attn_heads": 8,
17
  "attn_dim_head": 64,
18
  "sparse_attn": false,
19
  "memory_efficient": true
 
4
  "decoder": {
5
  "unets": [
6
  {
7
+ "dim": 512,
8
  "cond_dim": 512,
9
  "image_embed_dim": 768,
10
  "text_embed_dim": 768,
 
13
  "dim_mults": [1, 2, 3, 4],
14
  "self_attn": [false, true, true, true],
15
  "num_resnet_blocks": 4,
16
+ "attn_heads": 16,
17
  "attn_dim_head": 64,
18
  "sparse_attn": false,
19
  "memory_efficient": true