NagaSaiAbhinay commited on
Commit
d5aad48
·
1 Parent(s): 5a9279a

Make config match the code at tuneavideo port for diffusers

Browse files
Files changed (1) hide show
  1. unet/config.json +1 -8
unet/config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "UNet3DConditionModel",
3
- "_diffusers_version": "0.11.1",
4
  "act_fn": "silu",
5
  "attention_head_dim": 8,
6
  "block_out_channels": [
@@ -10,7 +10,6 @@
10
  1280
11
  ],
12
  "center_input_sample": false,
13
- "class_embed_type": null,
14
  "cross_attention_dim": 768,
15
  "down_block_types": [
16
  "CrossAttnDownBlock3D",
@@ -19,16 +18,11 @@
19
  "DownBlock3D"
20
  ],
21
  "downsample_padding": 1,
22
- "dual_cross_attention": false,
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
  "resnet_time_scale_shift": "default",
@@ -41,7 +35,6 @@
41
  ],
42
  "upcast_attention": false,
43
  "use_linear_projection": false,
44
- "conv_type": "inflated_conv_3d",
45
  "use_temporal_transformer": false,
46
  "use_temporal_conv": false,
47
  "sub_blocks_type": "3d"
 
1
  {
2
  "_class_name": "UNet3DConditionModel",
3
+ "_diffusers_version": "0.17.0.dev0",
4
  "act_fn": "silu",
5
  "attention_head_dim": 8,
6
  "block_out_channels": [
 
10
  1280
11
  ],
12
  "center_input_sample": false,
 
13
  "cross_attention_dim": 768,
14
  "down_block_types": [
15
  "CrossAttnDownBlock3D",
 
18
  "DownBlock3D"
19
  ],
20
  "downsample_padding": 1,
 
 
 
21
  "in_channels": 4,
22
  "layers_per_block": 2,
23
  "mid_block_scale_factor": 1,
 
24
  "norm_eps": 1e-05,
25
  "norm_num_groups": 32,
 
26
  "only_cross_attention": false,
27
  "out_channels": 4,
28
  "resnet_time_scale_shift": "default",
 
35
  ],
36
  "upcast_attention": false,
37
  "use_linear_projection": false,
 
38
  "use_temporal_transformer": false,
39
  "use_temporal_conv": false,
40
  "sub_blocks_type": "3d"