---
library_name: pytorch
tags:
- image-inpainting
- masked-image-modeling
- bidirectional-transformer
- custom-model
license: mit
---

# Better Inpainting Transformer

This repository contains the improved inpainting model.

## Files
- `pytorch_model.bin`: inference weights
- `config.json`: config
- `training_metadata.json`: epoch and val loss for best uploaded checkpoint
- `model_last.pt`: resumable training checkpoint
- `model_best.pt`: best local checkpoint

## Notes
Recreate the `BetterInpaintTransformer` class and load the state dict manually.