FLUX 3 Action DROID on RTX 5090
Optimized TensorRT runtime for the FLUX 3 Action DROID policy on NVIDIA GeForce RTX 5090, in two precisions: fp8 and bf16.
Requirements
- NVIDIA GeForce RTX 5090 (compute capability 12.0), TensorRT 11.3.0.99, CUDA 12.8, PyTorch with CUDA.
Test
python run.py --precision bf16 --inputs example/bf16/synthetic-inputs.safetensors --reference example/bf16/synthetic-reference.safetensors
python verify.py
Prepare real inputs from Black Forest Labs' released FLUX 3 Action DROID policy and checkpoint (FLUX Kommunity License). A synthetic example is included so the runtime can be tested without that download.
Licence
FLUX Kommunity License, non-commercial. See LICENSE.md and NOTICE.
- Downloads last month
- -