From e31eccd25924ebca68dbd8267d6fc34991c72b51 Mon Sep 17 00:00:00 2001
From: Piotr Lachert <piotr.lachert@surveily.com>
Date: Wed, 20 Nov 2024 19:56:18 +0100
Subject: [PATCH] Fix torch related compatibility issues

---
 src/surveily.developer.ai/base/Dockerfile | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/surveily.developer.ai/base/Dockerfile b/src/surveily.developer.ai/base/Dockerfile
index 5c54acd..0a7538b 100644
--- a/src/surveily.developer.ai/base/Dockerfile
+++ b/src/surveily.developer.ai/base/Dockerfile
@@ -53,9 +53,9 @@ USER vscode
 # - Runtime parameters
 RUN python3 -m pip install -U pip && \
     python3 -m pip install --user --quiet --no-cache-dir \
-    torch==2.0.0 \
-    torchvision==0.15.1 \
-    ultralytics==8.2.28 \
+    torch==2.5.1 \
+    torchvision==0.20.1 \
+    ultralytics==8.3.34 \
     huggingface_hub==0.18.0
 
 RUN python3 -m pip install --user --quiet --no-cache-dir \
@@ -90,7 +90,7 @@ RUN python3 -m pip install --user --quiet --no-cache-dir \
     nbconvert \
     xmltodict \
     autoflake \
-    lightning==2.3.3 \
+    lightning==2.4.0 \
     streamlit \
     pytest-cov \
     ipywidgets \