config.json

{
  "architectures": [
    "AlphaUNet"
  ],
  "model_type": "alpha-depth",
  "auto_map": {
    "AutoConfig": "model.AlphaDepthConfig",
    "AutoModel": "model.AlphaUNet"
  },
  "input_size": [3, 128, 128],
  "hidden_channels": 32,
  "transformers_version": "4.35.0"
}