BackTo2014 commited on
Commit
c2416d8
·
verified ·
1 Parent(s): 38ad182

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -6
config.json CHANGED
@@ -2,13 +2,11 @@
2
  "model_type": "UNet",
3
  "T": 1000,
4
  "channel": 128,
5
- "channel_mult": [1, 2, 2, 2],
6
- "attn": [1],
7
  "num_res_blocks": 2,
8
- "dropout": 0.1,
9
  "image_size": 32,
10
  "in_channels": 3,
11
- "out_channels": 3,
12
- "time_embedding_dim": 512, // 128 * 4
13
- "norm_num_groups": 32
14
  }
 
2
  "model_type": "UNet",
3
  "T": 1000,
4
  "channel": 128,
5
+ "channel_mult": [1, 2, 3, 4],
6
+ "attn": [2],
7
  "num_res_blocks": 2,
8
+ "dropout": 0.15,
9
  "image_size": 32,
10
  "in_channels": 3,
11
+ "out_channels": 3
 
 
12
  }