File size: 1,001 Bytes
15b623a
571e007
 
 
 
 
 
 
 
fcc45fd
571e007
07e9572
60d41a0
fcc45fd
571e007
 
fcc45fd
 
60d41a0
3f55302
 
ed12c3f
 
 
129c2fc
3f55302
15b623a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04

# Install Poetry
RUN apt-get update && apt-get install -y git curl python3 python3-pip build-essential ffmpeg libsm6 libxext6
RUN curl -sSL https://install.python-poetry.org | python3 -
ENV PATH="/root/.local/bin:${PATH}"
RUN poetry config virtualenvs.create false

# Install dependencies
WORKDIR /app

RUN git clone https://github.com/fishaudio/fish-diffusion.git --depth 1 --branch v2.1.0

WORKDIR /app/fish-diffusion
RUN poetry install

COPY --chown=user checkpoint.ckpt checkpoints/checkpoint.ckpt
COPY --chown=user exp_hifisinger.py configs/exp_hifisinger.py

ENV GRADIO_SERVER_NAME=0.0.0.0
ENV GRADIO_SERVER_PORT=7860
ENV NUMBA_CACHE_DIR=/tmp/numba
ENV TRANSFORMERS_CACHE=/tmp/huggingface
ENV MPLCONFIGDIR=/tmp/matplotlib
ENV TORCH_HOME=/tmp/torch

CMD python3 tools/hifisinger/inference.py --config configs/exp_hifisinger.py --checkpoint checkpoints/checkpoint.ckpt --speaker opencpop --max_slice_duration 120 --min_silence_duration 2 --gradio