jianzongwu
commited on
fix config error
Browse files- unet/config.json +4 -4
unet/config.json
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
{
|
2 |
"_class_name": "RoviModel",
|
3 |
-
"_diffusers_version": "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":
|
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 |
-
"
|
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 |
+
}
|