Add X-VC voice conversion mode
This commit is contained in:
@@ -0,0 +1,43 @@
|
||||
FROM nvidia/cuda:12.8.1-cudnn-runtime-ubuntu24.04
|
||||
|
||||
ARG XVC_COMMIT=49df8c591eafc48b096e466d96f9839f9c0dd739
|
||||
|
||||
RUN apt-get update \
|
||||
&& apt-get install -y --no-install-recommends \
|
||||
ca-certificates curl ffmpeg git libsndfile1 python3 python3-pip python3-venv \
|
||||
&& rm -rf /var/lib/apt/lists/*
|
||||
|
||||
RUN git clone https://github.com/Jerrister/X-VC.git /opt/xvc \
|
||||
&& cd /opt/xvc \
|
||||
&& git checkout "${XVC_COMMIT}" \
|
||||
&& rm -rf .git
|
||||
|
||||
RUN python3 -m venv /opt/venv
|
||||
ENV PATH="/opt/venv/bin:${PATH}"
|
||||
|
||||
# RTX 5080: use a CUDA 12.8 PyTorch build instead of X-VC's older training pin.
|
||||
RUN python -m pip install --no-cache-dir --upgrade pip \
|
||||
&& python -m pip install --no-cache-dir \
|
||||
--index-url https://download.pytorch.org/whl/cu128 \
|
||||
"torch==2.8.0" "torchaudio==2.8.0" \
|
||||
&& python -m pip install --no-cache-dir \
|
||||
"gradio>=5.49,<6" "huggingface_hub>=0.34,<2" \
|
||||
"transformers>=4.44,<5" "hydra-core>=1.3,<2" "omegaconf>=2.3,<3" \
|
||||
"x-transformers>=1.40,<3" "einops>=0.8,<1" "einx>=0.3,<1" \
|
||||
"numpy>=1.26,<3" "scipy>=1.13,<2" "soundfile>=0.12,<1" \
|
||||
"soxr>=0.5,<1" "tqdm>=4.66,<5" "wandb>=0.18,<1"
|
||||
|
||||
RUN python -m pip install --no-cache-dir "descript-audiotools==0.7.2"
|
||||
|
||||
COPY app.py /opt/xvc/local_webui.py
|
||||
COPY inference_log.py /opt/xvc/utils/log.py
|
||||
|
||||
ENV HF_HOME=/models/huggingface \
|
||||
PYTHONUNBUFFERED=1
|
||||
|
||||
EXPOSE 8009
|
||||
|
||||
HEALTHCHECK --interval=5s --timeout=3s --start-period=600s --retries=3 \
|
||||
CMD curl -fsS http://127.0.0.1:8009/ >/dev/null || exit 1
|
||||
|
||||
CMD ["python", "/opt/xvc/local_webui.py"]
|
||||
Reference in New Issue
Block a user