diff --git a/autogluon/training/docker/1.2/py3/Dockerfile.cpu b/autogluon/training/docker/1.2/py3/Dockerfile.cpu
index 7eddac42b103..2d3c382c507c 100644
--- a/autogluon/training/docker/1.2/py3/Dockerfile.cpu
+++ b/autogluon/training/docker/1.2/py3/Dockerfile.cpu
@@ -52,8 +52,8 @@ RUN pip install --no-cache-dir -U --trusted-host pypi.org --trusted-host files.p
  && pip install --no-cache-dir "pillow<11.0.0" \
  && pip install --no-cache-dir "ninja<1.11.1.1" \
  # Fix pip test
- && pip install --no-cache-dir "dill<0.3.9" \
- && pip install --no-cache-dir "multiprocess>0.70.16"
+ && pip install --no-cache-dir "datasets<3.1.0" \
+ && pip install --no-cache-dir "pathos<0.3.3"
 
 
 RUN HOME_DIR=/root \
diff --git a/autogluon/training/docker/1.2/py3/cu124/Dockerfile.gpu b/autogluon/training/docker/1.2/py3/cu124/Dockerfile.gpu
index 239d28590dbb..6fbf64d0c6f2 100644
--- a/autogluon/training/docker/1.2/py3/cu124/Dockerfile.gpu
+++ b/autogluon/training/docker/1.2/py3/cu124/Dockerfile.gpu
@@ -50,8 +50,8 @@ RUN pip install --no-cache-dir -U --trusted-host pypi.org --trusted-host files.p
  && pip install --no-cache-dir "pillow<11.0.0" \
  && pip install --no-cache-dir "ninja<1.11.1.1" \
   # Fix pip test
- && pip install --no-cache-dir "dill<0.3.9" \
- && pip install --no-cache-dir "multiprocess>0.70.16"
+&& pip install --no-cache-dir "datasets<3.1.0" \
+&& pip install --no-cache-dir "pathos<0.3.3"
 
 
 RUN HOME_DIR=/root \