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