checkpoint_path: https://dl.fbaipublicfiles.com/videoseal/pixelseal/checkpoint.pth
args:
  attenuation_config: configs/attenuation.yaml
  attenuation: jnd_1_1
  nbits: 256
  hidden_size_multiplier: 1.0
  img_size_proc: 256
  blending_method: additive
  scaling_w: 0.2
  scaling_i: 1.0
  videowam_chunk_size: 32
  videowam_step_size: 8
embedder:
  model: unet_base_yuv_quant
  params:
    msg_processor:
      nbits: 16
      hidden_size: 32
      msg_processor_type: 'binary+concat'
    unet:
      in_channels: 1
      out_channels: 1
      z_channels: 16
      num_blocks: 8
      activation: 'relu'
      normalization: 'batch'
      z_channels_mults: [2, 4, 8, 16]
      last_tanh: True
extractor:
  model: convnext_tiny
  params:
    encoder:
      depths: [3, 3, 9, 3]
      dims: [96, 192, 384, 768]
    pixel_decoder:
      pixelwise: False
      upscale_stages: [1]
      embed_dim: 768
      nbits: 16
      sigmoid_output: False