NagaSaiAbhinay commited on
Commit
0afbb4e
·
1 Parent(s): f29f022

Make it compatible with latest changes

Browse files

conv_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

Files changed (1) hide show
  1. 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
  }