diff --git a/Dockerfile.cpu b/Dockerfile.cpu index 0c3fa6c..ba83199 100644 --- a/Dockerfile.cpu +++ b/Dockerfile.cpu @@ -1,6 +1,6 @@ FROM python:3.10 -ARG FACEFUSION_VERSION=1.1.0 +ARG FACEFUSION_VERSION=1.2.0 ENV GRADIO_SERVER_NAME=0.0.0.0 WORKDIR /facefusion @@ -9,4 +9,4 @@ RUN apt-get update RUN apt-get install ffmpeg -y RUN git clone https://github.com/facefusion/facefusion.git --branch $FACEFUSION_VERSION --single-branch . -RUN pip install -r requirements.txt +RUN python install.py --onnxruntime cpu diff --git a/Dockerfile.cuda b/Dockerfile.cuda index af44d12..a88e7a1 100644 --- a/Dockerfile.cuda +++ b/Dockerfile.cuda @@ -1,6 +1,6 @@ FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04 -ARG FACEFUSION_VERSION=1.1.0 +ARG FACEFUSION_VERSION=1.2.0 ENV GRADIO_SERVER_NAME=0.0.0.0 WORKDIR /facefusion @@ -13,6 +13,4 @@ RUN apt-get install git -y RUN apt-get install ffmpeg -y RUN git clone https://github.com/facefusion/facefusion.git --branch $FACEFUSION_VERSION --single-branch . -RUN pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu118 -RUN pip uninstall onnxruntime -y -RUN pip install onnxruntime-gpu==1.15.1 +RUN python install.py --onnxruntime cuda diff --git a/Dockerfile.rocm b/Dockerfile.rocm index 6e18614..89f49cc 100644 --- a/Dockerfile.rocm +++ b/Dockerfile.rocm @@ -1,6 +1,6 @@ FROM rocm/dev-ubuntu-22.04:5.4.2-complete -ARG FACEFUSION_VERSION=1.1.0 +ARG FACEFUSION_VERSION=1.2.0 ARG ONNXRUNTIME_VERSION=1.15.0 ENV GRADIO_SERVER_NAME=0.0.0.0