diff --git a/poetry.lock b/poetry.lock
index f9cedb8484..6d8f1492a0 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1c6961ba0c2fbee0174d3bb7232794182710b65a145d4c4f1777f45c9413dc60
-size 635656
+oid sha256:d9fc1dff333ab867fcbc1e02676107f6c5a2d5d1d9f43327a4856bc873142418
+size 638741
diff --git a/pyproject.toml b/pyproject.toml
index 25b61eb50b..2963337733 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -164,7 +164,7 @@ redis = "^4.3.4"
 s2sphere = "^0.2.5"
 scikit-image = "^0.19.3"
 scikit-learn = "^1.1.1"
-segmentation-models-pytorch = "==0.2.1"
+segmentation-models-pytorch = "==0.3.2"
 simplejson = "^3.17.6"
 SQLAlchemy = "^1.4.39"
 torch = { url = "https://download.pytorch.org/whl/cu118/torch-2.0.0%2Bcu118-cp38-cp38-linux_x86_64.whl" }