storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-09 00:09:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

facefusion/comfy: run on cuda 12

Browse Source
This commit is contained in:
Kasisnu
2024-07-04 16:43:13 +05:30
parent ef25d44bbd
commit 35c2363f6e
2 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
workflows/comfy/Dockerfile
+3 -2
View File
@@ -4,7 +4,8 @@
# Nvidia does not provide a CUDA 11.6 with Ubuntu 22.04, which mirrors my own desktop machine. Their CUDA 11.6 + Ubuntu 20.04 image has
# a stale Python (3.8), so we have to use the DeadSnakes PPA to install a more modern version.
# Gross.
FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
# FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
FROM nvidia/cuda:12.2.2-cudnn8-devel-ubuntu22.04
# See: https://github.com/opencontainers/image-spec/blob/master/annotations.md
LABEL org.opencontainers.image.title='Storyteller ML: Voice Conversion'
@@ -71,7 +72,7 @@ RUN venv/bin/python -m pip install -r ComfyUI/requirements.txt --extra-index-url
COPY --chown=appuser:appgroup required_requirements.txt /app/required_requirements.txt
RUN venv/bin/python -m pip install -r required_requirements.txt --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ && \
venv/bin/python -m pip uninstall onnxruntime onnxruntime-gpu -y && \
venv/bin/python -m pip install onnxruntime-gpu ort-nightly-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ && \
venv/bin/python -m pip install onnxruntime-gpu==1.17.1 ort-nightly-gpu --index-url=https://pkgs.dev.azure.com/onnxruntime/onnxruntime/_packaging/onnxruntime-cuda-12/pypi/simple/ --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ && \
venv/bin/python -m pip cache purge
COPY --chown=appuser:appgroup repos.json /app/repos.json
workflows/comfy/facefusion-runner.sh
+2 -2
View File
@@ -49,5 +49,5 @@ deactivate || true
cd /facefusion
# Set up the environment
source venv/bin/activate
venv/bin/python3 -m pip install -r requirements.txt
venv/bin/python3 run.py -s $input_audio -t $input_video --frame-processors lip_syncer --frame-processors face_enhancer -o $output --headless
# venv/bin/python3 -m pip install -r requirements.txt
venv/bin/python3 run.py -s $input_audio -t $input_video --frame-processors lip_syncer --frame-processors face_enhancer --execution-provider cuda -o $output --headless