From 0c5f56c8a6a384d45f65035faf23fa23ecc80590 Mon Sep 17 00:00:00 2001 From: Siwat Sirichai Date: Mon, 10 Nov 2025 00:35:46 +0700 Subject: [PATCH] add onnx requirements --- requirements.txt | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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