From 416d43891b8770271544ecc17a9a79610a15a19b Mon Sep 17 00:00:00 2001 From: Amartya Deshmukh Date: Wed, 10 Sep 2025 12:01:16 +0530 Subject: [PATCH] Update requirements-gpu.txt --- requirements-gpu.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/requirements-gpu.txt b/requirements-gpu.txt index 75dd998..f5bd746 100644 --- a/requirements-gpu.txt +++ b/requirements-gpu.txt @@ -5,7 +5,8 @@ onnx>=1.14.0 protobuf>=3.20.2 opencv-python pandas +albumentations==1.3.1 pydantic safetensors onnxruntime>=1.15.0 -onnxruntime-gpu>=1.15.0 \ No newline at end of file +onnxruntime-gpu>=1.15.0