BackTo2014 commited on
Commit
baf3d19
1 Parent(s): 9475560

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +18 -18
config.json CHANGED
@@ -1,33 +1,33 @@
1
  {
2
- "in_channels": 3, // 输入通道数
3
- "out_channels": 3, // 输出通道数
4
- "down_block_types": [ // 下采样块类型
5
  "DownBlock2D",
6
  "AttnDownBlock2D",
7
  "DownBlock2D"
8
  ],
9
- "up_block_types": [ // 上采样块类型
10
  "UpBlock2D",
11
  "AttnUpBlock2D",
12
  "UpBlock2D"
13
  ],
14
- "block_out_channels": [ // 每个下采样阶段的输出通道数
15
  128,
16
  256,
17
  384,
18
  512
19
  ],
20
- "layers_per_block": 2, // 每个残差块内的层数
21
- "norm_num_groups": 32, // 归一化层的组数
22
- "cross_attention_dim": null, // 交叉注意力维度(未使用)
23
- "attention_head_dim": 8, // 注意力头的维度
24
- "resnet_time_scale_shift": "default", // 时间尺度变换类型
25
- "class_embed_type": null, // 类别嵌入类型(未使用)
26
- "num_class_embeds": null, // 类别嵌入数量(未使用)
27
- "use_linear_projection": false, // 是否使用线性投影(未使用)
28
- "only_cross_attention": false, // 是否仅使用交叉注意力(未使用)
29
- "mid_block_scale_factor": 1, // 中间块的尺度因子
30
- "out_bias": true, // 输出是否带有偏置项
31
- "flip_sin_to_cos": true, // 是否翻转正弦到余弦(时间编码)
32
- "freq_shift": 0 // 频率偏移量(时间编码)
33
  }
 
1
  {
2
+ "in_channels": 3,
3
+ "out_channels": 3,
4
+ "down_block_types": [
5
  "DownBlock2D",
6
  "AttnDownBlock2D",
7
  "DownBlock2D"
8
  ],
9
+ "up_block_types": [
10
  "UpBlock2D",
11
  "AttnUpBlock2D",
12
  "UpBlock2D"
13
  ],
14
+ "block_out_channels": [
15
  128,
16
  256,
17
  384,
18
  512
19
  ],
20
+ "layers_per_block": 2,
21
+ "norm_num_groups": 32,
22
+ "cross_attention_dim": null,
23
+ "attention_head_dim": 8,
24
+ "resnet_time_scale_shift": "default",
25
+ "class_embed_type": null,
26
+ "num_class_embeds": null,
27
+ "use_linear_projection": false,
28
+ "only_cross_attention": false,
29
+ "mid_block_scale_factor": 1,
30
+ "out_bias": true,
31
+ "flip_sin_to_cos": true,
32
+ "freq_shift": 0
33
  }