birefnet-tensorrt / README.md
carpedm20's picture
Upload README.md with huggingface_hub
c479cbf verified

BiRefNet TensorRT Engine

Pre-built TensorRT engine for BiRefNet background removal.

Specifications

Setting Value
Resolution 1024x1024
Batch Size 1-4
Precision fp16
TensorRT 10.14.1.48
Built on NVIDIA GeForce RTX 4090
Engine Size 505.64 MB

Usage

The engine is automatically downloaded when using BiRefNet with TensorRT enabled.

from ai.models.birefnet import BiRefNetSegmentationModel

model = BiRefNetSegmentationModel(use_tensorrt=True)
result = await model.run(image, payload=payload)

Rebuild

make run-ai-local -- python -m scripts.tensorrt.birefnet --force

Built at: 2026-01-22 07:46:24 UTC