README.md

metadata

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 format from the original .pt release by Sanster.

• ✅ 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

How to use

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)