Sapiens2 TensorRT Engine
Pre-built single-engine TensorRT artifact for facebook/sapiens2-seg-0.4b.
Specifications
| Setting | Value |
|---|---|
| Variant | 0.4b |
| Revision | h100-sm90-trt10.14.1.48-max-batch-1-fp8-default-normqk-exp |
| Precision | fp8 |
| Output kind | mask |
| Output dtype | uint8 |
| Input shape | [1, 3, 1024, 768] |
| Max batch size | 1 |
| TensorRT | 10.14.1.48 |
| CUDA | 13.1 |
| Built on | h100-sm90 |
| Engine size | 780.51 MB |
| Model engine | sapiens2_0.4b_seg_fp8_mask_uint8.engine |
Usage
Fury downloads this engine when runtime="tensorrt" and
SAPIENS2_TENSORRT_ENGINE_REVISION=h100-sm90-trt10.14.1.48-max-batch-1-fp8-default-normqk-exp.
from ai.models.segmentation.sapiens2 import Sapiens2SegmentationModel
model = Sapiens2SegmentationModel(
model_name="0.4b",
runtime="tensorrt",
engine_revision="h100-sm90-trt10.14.1.48-max-batch-1-fp8-default-normqk-exp",
)
Local Download
make cmd-ai-local -- python -m scripts.tensorrt.sapiens2 download \
--model-name 0.4b \
--engine-revision h100-sm90-trt10.14.1.48-max-batch-1-fp8-default-normqk-exp
Built at: 2026-07-04 10:08:28 UTC
- Downloads last month
- -
Model tree for carpedm20/sapiens2-seg-tensorrt
Base model
facebook/sapiens2-pretrain-0.4b Finetuned
facebook/sapiens2-seg-0.4b