birefnet-tensorrt / README.md
carpedm20's picture
Upload README.md with huggingface_hub
c479cbf verified
# BiRefNet TensorRT Engine
Pre-built TensorRT engine for [BiRefNet](ZhengPeng7/BiRefNet_dynamic) 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.
```python
from ai.models.birefnet import BiRefNetSegmentationModel
model = BiRefNetSegmentationModel(use_tensorrt=True)
result = await model.run(image, payload=payload)
```
## Rebuild
```bash
make run-ai-local -- python -m scripts.tensorrt.birefnet --force
```
Built at: 2026-01-22 07:46:24 UTC