diff --git a/setup.py b/setup.py
index b65cd4c..203b244 100644
--- a/setup.py
+++ b/setup.py
@@ -36,7 +36,7 @@
         'scikit-learn',
         'scipy',
         'segmentation_models_pytorch',
-        'torch',
+        'torch<1.13',
         'torchinfo',
         'torchplot',
         'torchvision',