🚀 What is inference-models?
inference-models is the library to make predictions from computer vision models provided by Roboflow — designed to
be fast, reliable, and user-friendly. It offers:
- Multi-Backend Support: Run models with PyTorch, ONNX, TensorRT, or Hugging Face backends
- Automatic Model Loading: Smart model resolution and backend selection
- Minimal Dependencies: Composable extras system for installing only what you need
- Behavior-Based Interfaces: Models with similar behavior share consistent APIs; custom models can define their own
- Full Roboflow Platform Support: Run any model trained on Roboflow
Visit our documentation for more information.
🛣️ Roadmap
With release 0.19.0, we have reached the first stable release of inference-models and fully integrated
the package to inference - our main inference package, making it selectable backend for running predictions
from models.
We are still making changes to add new features and models. API should be fairly stable already, but the problems may still occur. If you encounter any issues, please report them.
💻 Installation
CPU installation:
uv pip install inference-models
# or with pip
pip install inference-models
inference-models can be installed with CUDA and TensorRT support - see Installation Guide for more options.
🏃➡️ Usage
Pretrained Models
Load and run a pretrained model:
import cv2
import supervision as sv
from inference_models import AutoModel
# Load pretrained model from Roboflow
model = AutoModel.from_pretrained("rfdetr-base")
# Run inference (works with numpy arrays or torch.Tensor)
image = cv2.imread("<path-to-your-image>")
predictions = model(image)
# Use with supervision
annotator = sv.BoxAnnotator()
annotated = annotator.annotate(image, predictions[0].to_supervision())
Your Roboflow Models
Load and run models trained on the Roboflow platform:
import cv2
import supervision as sv
from inference_models import AutoModel
# Load your custom model from Roboflow
model = AutoModel.from_pretrained(
"<your-project>/<version>",
api_key="<your-api-key>" # model access secured with API key
)
# Run inference (works with numpy arrays or torch.Tensor)
image = cv2.imread("<path-to-your-image>")
predictions = model(image)
# Use with supervision
annotator = sv.BoxAnnotator()
annotated = annotator.annotate(image, predictions[0].to_supervision())
🧠 Supported Model Architectures
- RFDetr
- SAM models family
- Vision-Language Models (Florence, PaliGemma, Qwen, SmolVLM, Moondream)
- OCR (DocTR, EasyOCR, TrOCR)
- YOLO
- and many more
For detailed model documentation, see Supported Models.
🔧 Run your local models
Load your own model implementations from a local directory - models with architectures not in the main inference-models package. This is especially valuable for production deployment of custom models.
from inference_models import AutoModel
model = AutoModel.from_pretrained(
"/path/to/my_custom_model",
allow_local_code_packages=True
)
See Load Models from Local Packages for complete details on creating custom model packages.
📄 License
The inference-models package is licensed under Apache 2.0. Individual models may have different licenses - see the Supported Models for details.
Ready to get started? Head to the Quick Overview →
1 maintainer
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file inference_models-0.32.0.tar.gz.
File metadata
- Download URL: inference_models-0.32.0.tar.gz
- Upload date:
- Size: 1.8 MB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via:
twine/6.1.0 CPython/3.13.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
2feb4fef46b0d259c5923c0b01748b1ad849f689c77efb9379fad1f7cf6ec802
|
|
| MD5 |
e961caeac537007c5472b640c4a0662f
|
|
| BLAKE2b-256 |
820d1163537598e18be2304bb28c6866ccd2c7c023d1399bfb09d27e6955987b
|
Provenance
The following attestation bundles were made for inference_models-0.32.0.tar.gz:
Publisher:
publish.pypi.inference_exp.yml on roboflow/inference
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
inference_models-0.32.0.tar.gz -
Subject digest:
2feb4fef46b0d259c5923c0b01748b1ad849f689c77efb9379fad1f7cf6ec802 - Sigstore transparency entry: 2212223734
- Sigstore integration time:
-
Permalink:
roboflow/inference@7302af354f7328037963de5103289ae91f95f642 -
Branch / Tag:
refs/heads/peter/cosmos3-generative - Owner: https://github.com/roboflow
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
self-hosted -
Publication workflow:
publish.pypi.inference_exp.yml@7302af354f7328037963de5103289ae91f95f642 -
Trigger Event:
workflow_dispatch
-
Statement type:
File details
Details for the file inference_models-0.32.0-py3-none-any.whl.
File metadata
- Download URL: inference_models-0.32.0-py3-none-any.whl
- Upload date:
- Size: 2.0 MB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via:
twine/6.1.0 CPython/3.13.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
67f24a6b3c26e1ef617132e3f79cf9db3dbc0e974972c6b7a0babb3e943abac3
|
|
| MD5 |
da3a588c62977b657a20d27be6632aee
|
|
| BLAKE2b-256 |
a2ffae43c56ac2631e613e2913d203e9802d2cb5201ea1c85aa436cc82b9dc44
|
Provenance
The following attestation bundles were made for inference_models-0.32.0-py3-none-any.whl:
Publisher:
publish.pypi.inference_exp.yml on roboflow/inference
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
inference_models-0.32.0-py3-none-any.whl -
Subject digest:
67f24a6b3c26e1ef617132e3f79cf9db3dbc0e974972c6b7a0babb3e943abac3 - Sigstore transparency entry: 2212223745
- Sigstore integration time:
-
Permalink:
roboflow/inference@7302af354f7328037963de5103289ae91f95f642 -
Branch / Tag:
refs/heads/peter/cosmos3-generative - Owner: https://github.com/roboflow
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
self-hosted -
Publication workflow:
publish.pypi.inference_exp.yml@7302af354f7328037963de5103289ae91f95f642 -
Trigger Event:
workflow_dispatch
-
Statement type:
