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 && \ cd fish-diffusion && \ git checkout ae5689c2cb6cae5ac439b4bd5e88be015d89d060 WORKDIR /app/fish-diffusion RUN poetry install COPY --chown=user . . 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/gradio_ui.py