patrickvonplaten commited on
Commit
d417829
1 Parent(s): 8c80146

Update unet/config.json

Browse files
Files changed (1) hide show
  1. unet/config.json +2 -18
unet/config.json CHANGED
@@ -10,8 +10,6 @@
10
  1280,
11
  1280
12
  ],
13
- "center_input_sample": false,
14
- "class_embed_type": null,
15
  "cross_attention_dim": 1024,
16
  "down_block_types": [
17
  "CrossAttnDownBlock3D",
@@ -19,30 +17,16 @@
19
  "CrossAttnDownBlock3D",
20
  "DownBlock3D"
21
  ],
22
- "downsample_padding": 1,
23
- "flip_sin_to_cos": true,
24
- "freq_shift": 0,
25
  "in_channels": 4,
26
  "layers_per_block": 2,
27
- "mid_block_scale_factor": 1,
28
- "mid_block_type": "UNetMidBlock3DCrossAttn",
29
  "norm_eps": 1e-05,
30
  "norm_num_groups": 32,
31
- "num_class_embeds": null,
32
- "only_cross_attention": false,
33
  "out_channels": 4,
34
- "projection_class_embeddings_input_dim": null,
35
- "resnet_time_scale_shift": "default",
36
  "sample_size": 32,
37
- "time_cond_proj_dim": null,
38
- "time_embedding_type": "positional",
39
- "timestep_post_act": null,
40
  "up_block_types": [
41
  "UpBlock3D",
42
  "CrossAttnUpBlock3D",
43
  "CrossAttnUpBlock3D",
44
  "CrossAttnUpBlock3D"
45
- ],
46
- "upcast_attention": false,
47
- "use_linear_projection": true
48
- }
 
10
  1280,
11
  1280
12
  ],
 
 
13
  "cross_attention_dim": 1024,
14
  "down_block_types": [
15
  "CrossAttnDownBlock3D",
 
17
  "CrossAttnDownBlock3D",
18
  "DownBlock3D"
19
  ],
 
 
 
20
  "in_channels": 4,
21
  "layers_per_block": 2,
 
 
22
  "norm_eps": 1e-05,
23
  "norm_num_groups": 32,
 
 
24
  "out_channels": 4,
 
 
25
  "sample_size": 32,
 
 
 
26
  "up_block_types": [
27
  "UpBlock3D",
28
  "CrossAttnUpBlock3D",
29
  "CrossAttnUpBlock3D",
30
  "CrossAttnUpBlock3D"
31
+ ]
32
+ }