diff --git a/services/reforge/Dockerfile b/services/reforge/Dockerfile
index 30332ce..e9845cf 100644
--- a/services/reforge/Dockerfile
+++ b/services/reforge/Dockerfile
@@ -60,6 +60,7 @@ RUN apt-get -y install libgoogle-perftools-dev && apt-get clean
 ENV LD_PRELOAD=libtcmalloc.so
 
 COPY . /docker
+RUN pip install --upgrade pydantic
 
 RUN \
   # mv ${ROOT}/style.css ${ROOT}/user.css && \
@@ -72,4 +73,4 @@ ENV NVIDIA_VISIBLE_DEVICES=all
 ENV CLI_ARGS=""
 EXPOSE 7860
 ENTRYPOINT ["/docker/entrypoint.sh"]
-CMD python -u webui.py --listen --port 7860 ${CLI_ARGS}
\ No newline at end of file
+CMD python -u webui.py --listen --port 7860 ${CLI_ARGS}