| --- |
| 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. |
