diff --git a/requirements.txt b/requirements.txt index e45d119..c580c8e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,4 +6,6 @@ av cuda-python nvidia-nvimgcodec-cu12 # GPU-accelerated JPEG encoding/decoding with nvJPEG python-dotenv # Load environment variables from .env file -torch_tensorrt \ No newline at end of file +torch_tensorrt +onnxslim>=0.1.71 +onnxruntime-gpu \ No newline at end of file