Model reference · open weights

sapiens-pretrain-torchscript

Available as managed deployment Licence fee Embeddings facebook Image embed 1 variants 1k dl/mo

sapiens-pretrain-torchscript is an open-weight embedding model from facebook. AxForge deploys and operates it for you on dedicated EU-owned hardware — with the licence handled where one is required.

Available as managed deployment — configured and operated for you on dedicated EU hardware, quoted per deployment.

What it is

Released byMeta
Published underfacebook
TypeEmbedding models
TaskImage embed
Runs withsapiens
Released2024-09-09
Popularity1k downloads / month
LicenceCommercial licence needed

About

What sapiens-pretrain-torchscript is

Model Details

Sapiens is a family of vision transformers pretrained on 300 million human images at 1024 x 1024 image resolution. The pretrained models, when finetuned for human-centric vision tasks, generalize to in-the-wild conditions. Sapiens-1B natively support 1K high-resolution inference. The resulting models exhibit remarkable generalization to in-the-wild data, even when labeled data is scarce or entirely synthetic.

Read the full model card
  • Developed by: Meta
  • Model type: Vision Transformer
  • License: Creative Commons Attribution-NonCommercial 4.0
  • Task: pretrain
  • Format: torchscript
  • File: sapiens_1b_epoch_173_torchscript.pt2

Model Card

  • Image Size: 1024 x 1024
  • Num Parameters: 1.169 B
  • FLOPs: 4.647 TFLOPs
  • Patch Size: 16 x 16
  • Embedding Dimensions: 1536
  • Num Layers: 40
  • Num Heads: 24
  • Feedforward Channels: 6144

More Resources

Uses

Pretrained 1B model can be used for feature extraction, fine-tuning, or as a starting point for training new models.

From the published model card. Full card on the HuggingFace links in the sidebar.

How it works

How embedding models work

Your textsentence / documentEncodermaps meaningVectorlist of numbersAn embedding model turns text into a vector, so similar meanings sit close together — the basis of search and RAG.

Using it via the API

Call it like any OpenAI endpoint

Once AxForge deploys sapiens-pretrain-torchscript for you, it answers on the OpenAI-compatible API — the same base URL and keys as every other model. (sapiens-pretrain-torchscript below is illustrative; you get the exact model name on deployment.)

$ curl -sS https://api.axforge.ai/v1/embeddings \
  -H "Authorization: Bearer $AXFORGE_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{"model":"sapiens-pretrain-torchscript","input":"text to embed"}'

Create an account — your API key is available in the console. 3M free tokens every 30 days with every new account.

© 2026 AxForge · EU-hosted AI infrastructure Pricing Docs Trust Privacy Terms