diff --git a/requirements.txt b/requirements.txt index 103dce8..c354c4c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,8 +1,8 @@ --extra-index-url https://download.pytorch.org/whl/cu124 albumentations==2.0.5 -lightning==2.5.0 +lightning==2.5.1 onnx==1.17.0 -onnxruntime==1.20.1 +onnxruntime==1.21.0 pytorch-msssim==1.0.0 torch==2.6.0 torchdata==0.11.0