| license: mit | |
| tags: | |
| - image-inpainting | |
| - LaMa | |
| - PyTorch | |
| - safetensors | |
| library_name: none | |
| # Big-LaMa Inpainting Model (Safetensors) | |
| This is the `big-lama` image inpainting model, converted to the [safetensors](https://github.com/huggingface/safetensors) format from the original `.pt` release by [Sanster](https://github.com/Sanster/models). | |
| - ✅ Only the `generator` weights are included | |
| - ❌ Evaluation code and FID scoring models have been removed | |
| - ✅ TorchScript converted to state_dict | |
| - ✅ 100% compatible with inference pipelines based on [LaMa](https://github.com/advimman/lama) | |
| ## How to use | |
| ```python | |
| from safetensors.torch import load_file | |
| from huggingface_hub import hf_hub_download | |
| # Download from Hub | |
| weights_path = hf_hub_download("michaelgold/big-lama-safetensors", "big-lama.safetensors") | |
| # Load into your model | |
| state_dict = load_file(weights_path) | |