NagaSaiAbhinay
commited on
Commit
·
0afbb4e
1
Parent(s):
f29f022
Make it compatible with latest changes
Browse filesconv_type: str = None, #Use inflated_conv_3d for TuneAVideo,
use_temporal_transformer: bool = True, #Use false for TuneAVideo
use_temporal_conv:bool = True, #Use false for TuneAVideo
sub_blocks_type: str = '2d' #Use 2d for ResNet2D, Transformer2D blocks in DownBlock3D. This is in TuneAVideo
- unet/config.json +5 -1
unet/config.json
CHANGED
@@ -40,5 +40,9 @@
|
|
40 |
"CrossAttnUpBlock3D"
|
41 |
],
|
42 |
"upcast_attention": false,
|
43 |
-
"use_linear_projection": false
|
|
|
|
|
|
|
|
|
44 |
}
|
|
|
40 |
"CrossAttnUpBlock3D"
|
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",
|
48 |
}
|