diff --git a/Dockerfile.cpu b/Dockerfile.cpu
index 722da9d..733941b 100644
--- a/Dockerfile.cpu
+++ b/Dockerfile.cpu
@@ -1,6 +1,6 @@
 FROM python:3.10
 
-ARG FACEFUSION_VERSION=2.3.0
+ARG FACEFUSION_VERSION=2.4.0
 ENV GRADIO_SERVER_NAME=0.0.0.0
 
 WORKDIR /facefusion
@@ -10,4 +10,4 @@ RUN apt-get install curl -y
 RUN apt-get install ffmpeg -y
 
 RUN git clone https://github.com/facefusion/facefusion.git --branch ${FACEFUSION_VERSION} --single-branch .
-RUN python install.py --torch cpu --onnxruntime default --skip-venv
+RUN python install.py --onnxruntime default --skip-venv
diff --git a/Dockerfile.cuda b/Dockerfile.cuda
index 76cb9b4..d1f9923 100644
--- a/Dockerfile.cuda
+++ b/Dockerfile.cuda
@@ -1,6 +1,6 @@
 FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
 
-ARG FACEFUSION_VERSION=2.3.0
+ARG FACEFUSION_VERSION=2.4.0
 ENV GRADIO_SERVER_NAME=0.0.0.0
 
 WORKDIR /facefusion
@@ -14,6 +14,6 @@ RUN apt-get install curl -y
 RUN apt-get install ffmpeg -y
 
 RUN git clone https://github.com/facefusion/facefusion.git --branch ${FACEFUSION_VERSION} --single-branch .
-RUN python install.py --torch cuda-11.8 --onnxruntime cuda-11.8 --skip-venv
+RUN python install.py --onnxruntime cuda-11.8 --skip-venv
 
 RUN cd /usr/local/lib/python3.10/dist-packages/torch/lib && ln -s libnvrtc-672ee683.so.11.2 libnvrtc.so
diff --git a/Dockerfile.rocm b/Dockerfile.rocm
index 608c026..0141ad8 100644
--- a/Dockerfile.rocm
+++ b/Dockerfile.rocm
@@ -1,6 +1,6 @@
 FROM rocm/dev-ubuntu-22.04:5.6
 
-ARG FACEFUSION_VERSION=2.3.0
+ARG FACEFUSION_VERSION=2.4.0
 ENV GRADIO_SERVER_NAME=0.0.0.0
 
 WORKDIR /facefusion
@@ -14,4 +14,4 @@ RUN apt-get install curl -y
 RUN apt-get install ffmpeg -y
 
 RUN git clone https://github.com/facefusion/facefusion.git --branch ${FACEFUSION_VERSION} --single-branch .
-RUN python install.py --torch rocm-5.6 --onnxruntime rocm-5.6 --skip-venv
+RUN python install.py --onnxruntime rocm-5.6 --skip-venv