diff --git a/Dockerfile.cpu b/Dockerfile.cpu index 817f935..4b19d9a 100644 --- a/Dockerfile.cpu +++ b/Dockerfile.cpu @@ -1,6 +1,6 @@ FROM python:3.10 -ARG FACEFUSION_VERSION=2.6.1 +ARG FACEFUSION_VERSION=3.0.0 ENV GRADIO_SERVER_NAME=0.0.0.0 WORKDIR /facefusion diff --git a/Dockerfile.cuda b/Dockerfile.cuda index 17dd54b..c07979d 100644 --- a/Dockerfile.cuda +++ b/Dockerfile.cuda @@ -1,6 +1,6 @@ -FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04 +FROM nvidia/cuda:12.6.1-cudnn-runtime-ubuntu22.04 -ARG FACEFUSION_VERSION=2.6.1 +ARG FACEFUSION_VERSION=3.0.0 ENV GRADIO_SERVER_NAME=0.0.0.0 WORKDIR /facefusion @@ -14,5 +14,5 @@ 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 --onnxruntime cuda-11.8 --skip-conda +RUN python install.py --onnxruntime cuda --skip-conda diff --git a/Dockerfile.rocm b/Dockerfile.rocm index 36bb645..d138c4f 100644 --- a/Dockerfile.rocm +++ b/Dockerfile.rocm @@ -1,6 +1,6 @@ -FROM rocm/dev-ubuntu-22.04:5.6 +FROM rocm/dev-ubuntu-22.04:6.2 -ARG FACEFUSION_VERSION=2.6.1 +ARG FACEFUSION_VERSION=3.0.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 --onnxruntime rocm-5.6 --skip-conda +RUN python install.py --onnxruntime rocm --skip-conda diff --git a/README.md b/README.md index 02163fc..8cc2290 100644 --- a/README.md +++ b/README.md @@ -1,7 +1,7 @@ FaceFusion Docker ================= -> Next generation face swapper and enhancer. +> Industry leading face manipulation platform. [![Build Status](https://img.shields.io/github/actions/workflow/status/facefusion/facefusion-docker/ci.yml.svg?branch=master)](https://github.com/facefusion/facefusion-docker/actions?query=workflow:ci) ![License](https://img.shields.io/badge/license-MIT-green)