Hugging Face's logo

nousr
/
dalle2-pytorch

Model card Files Community
nousr commited on
Commit
af7ee78
·
1 Parent(s): a12e4aa

Upload decoder/decoder_config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. decoder/decoder_config.json +2 -2
decoder/decoder_config.json CHANGED
@@ -3,13 +3,13 @@
3
  "decoder": {
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": 8,
15
  "attn_dim_head": 64,
 
3
  "decoder": {
4
  "unets": [
5
  {
6
+ "dim": 384,
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, 2, 4],
13
  "num_resnet_blocks": 4,
14
  "attn_heads": 8,
15
  "attn_dim_head": 64,