propainter / config.json
ruffy369's picture
Update config.json
7763a16 verified
{
"_name_or_path": "ruffy369/propainter",
"adversarial_weight": 0.01,
"architectures": [
"ProPainterModel"
],
"channels": [
64,
96,
128
],
"correlation_levels": 4,
"correlation_radius": 4,
"dropout": 0.0,
"flow_weight_flow_complete_net": 0.25,
"gan_loss": "hinge",
"hidden_size": 512,
"hole_weight": 1.0,
"in_channels": [
64,
64,
96
],
"initializer_range": 0.02,
"interp_mode": "nearest",
"kernel_size": [
7,
7
],
"kernel_size_3d": [
1,
3,
3
],
"kernel_size_3d_discriminator": [
3,
5,
5
],
"model_type": "propainter",
"neighbor_length": 10,
"no_dis": false,
"norm_fn": [
"batch",
"group",
"instance",
"none"
],
"num_attention_heads": 4,
"num_channels": 128,
"num_hidden_layers": 8,
"num_local_frames_propainter": 10,
"padding": 1,
"padding_inpaint_generator": [
3,
3
],
"patch_size": 3,
"perceptual_weight": 0.0,
"pool_size": [
4,
4
],
"raft_iter": 20,
"ref_stride": 10,
"stride": [
3,
3
],
"stride_3d": [
1,
1,
1
],
"strides": [
1,
2,
2
],
"subvideo_length": 80,
"transformers_version": "4.46.0.dev0",
"valid_weight": 1.0,
"window_size": [
5,
9
]
}