jianzongwu commited on
Commit
acb0eee
·
verified ·
1 Parent(s): 84de076

fix config error

Browse files
Files changed (1) hide show
  1. unet/config.json +4 -4
unet/config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "RoviModel",
3
- "_diffusers_version": "0.20.0",
4
  "act_fn": "silu",
5
  "attention_head_dim": 8,
6
  "block_out_channels": [
@@ -23,7 +23,7 @@
23
  "dual_cross_attention": false,
24
  "flip_sin_to_cos": true,
25
  "freq_shift": 0,
26
- "in_channels": 4,
27
  "lan_cross_attn": false,
28
  "layers_per_block": 2,
29
  "mask_decoder_type": "MaskDecoder",
@@ -37,7 +37,7 @@
37
  "ref_frame_type_2": null,
38
  "resnet_time_scale_shift": "default",
39
  "sample_size": 64,
40
- "st_attn": false,
41
  "up_block_types": [
42
  "UpBlock3D",
43
  "CrossAttnUpBlock3D",
@@ -46,4 +46,4 @@
46
  ],
47
  "upcast_attention": false,
48
  "use_linear_projection": false
49
- }
 
1
  {
2
  "_class_name": "RoviModel",
3
+ "_diffusers_version": "0.21.4",
4
  "act_fn": "silu",
5
  "attention_head_dim": 8,
6
  "block_out_channels": [
 
23
  "dual_cross_attention": false,
24
  "flip_sin_to_cos": true,
25
  "freq_shift": 0,
26
+ "in_channels": 8,
27
  "lan_cross_attn": false,
28
  "layers_per_block": 2,
29
  "mask_decoder_type": "MaskDecoder",
 
37
  "ref_frame_type_2": null,
38
  "resnet_time_scale_shift": "default",
39
  "sample_size": 64,
40
+ "temp_attn": true,
41
  "up_block_types": [
42
  "UpBlock3D",
43
  "CrossAttnUpBlock3D",
 
46
  ],
47
  "upcast_attention": false,
48
  "use_linear_projection": false
49
+ }