config.json

{
  "model_type": "dispvit",
  "architectures": [
    "DispViT"
  ],
  "encoder_type": "vitg",
  "groups": 4,
  "init_weights": false,
  "checkpoint_filename": "model_final.pt"
}