From cd4b10c8329516a2bb442d94a550ebe4604e90b8 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Sun, 16 Feb 2025 10:54:01 +0100 Subject: [PATCH] Fix imports and update dependencies --- requirements.txt | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/requirements.txt b/requirements.txt index b04959e..722121a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,10 @@ -lightning==2.4.0 +--extra-index-url https://download.pytorch.org/whl/cu124 +lightning==2.5.0 numpy==1.26.4 onnx==1.17.0 -onnxruntime==1.20.0 -opencv-python==4.10.0.84 +onnxruntime==1.20.1 +opencv-python==4.11.0.86 mxnet==1.9.1 pytorch-msssim==1.0.0 +torch==2.6.0 +torchvision==0.21.0