diff --git a/preconfigured_models/image_to_image/polygon_segmentation/requirements.txt b/preconfigured_models/image_to_image/polygon_segmentation/requirements.txt
index 19868d1..df95d16 100644
--- a/preconfigured_models/image_to_image/polygon_segmentation/requirements.txt
+++ b/preconfigured_models/image_to_image/polygon_segmentation/requirements.txt
@@ -1,5 +1,5 @@
 ultralytics==8.2.34
 mlflow==2.17.0
-torch==2.3.0
+torch==2.6.0
 torchvision==0.18
 dagshub
\ No newline at end of file