The model expects both the image and trimap (concatenated) as input. Use [ViTMatteImageProcessor] for this purpose.

VitMatteConfig
[[autodoc]] VitMatteConfig
VitMatteImageProcessor
[[autodoc]] VitMatteImageProcessor
    - preprocess
VitMatteForImageMatting
[[autodoc]] VitMatteForImageMatting
    - forward