config.json

{
  "_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
  ]
}