---
license: mit
base_model:
- facebook/sapiens
---

# Sapiens Exported Model (Schema 7.3)

This repository provides a re-exported checkpoint of the [facebook/sapiens](https://huggingface.co/facebook/sapiens) segmentation model using **PyTorch 2.5.1**, ensuring compatibility with **modern `torch.export.load()` workflows**.

---

## Background

The original SAPIENS checkpoints were exported in PyTorch 2.1.x and use **IR schema version `5.1`**, which causes `torch.export.load()` to fail on newer PyTorch versions (e.g., 2.2+), due to a mismatch in how versioning is handled internally.

Many users encounter the following error:

`ValueError: invalid literal for int() with base 10: b'5.1'`


To address this, we provide a **re-exported checkpoint** using **PyTorch 2.5.1**, which uses **schema version `7.3`**, fully compatible with current and future versions of PyTorch.

---

## Contents

- `..._bfloat16.pt2`: Re-exported IR checkpoint
- Compatible with: `torch.export.load()` in **PyTorch ≥ 2.3.0**
- Schema version: **7.3**

---

## How to Load

```python
from torch.export import load
from huggingface_hub import hf_hub_download

model_path = hf_hub_download("RyanL22/sapiens-bfloat16", "pose/checkpoints/sapiens_1b_goliath_best_goliath_AP_639_bfloat16.pt2")
model = load(model_path).module()
```

🔧 Make sure you are using PyTorch 2.3.0 or higher to ensure schema 7.x compatibility.

Credits
Original model: facebook/sapiens

Re-exported by: @RyanL22