FROM git.modelhub.org.cn:9443/enginex-cambricon/mlu370-pytorch:v25.01-torch2.5.0-torchmlu1.24.1-ubuntu22.04-py310

WORKDIR /workspace

ENV PATH=/torch/venv3/pytorch_infer/bin:/workspace/ffmpeg-mlu-v4.2.0/install/bin:/usr/local/neuware/bin:/usr/local/openmpi/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
RUN sed -i 's|source /torch/venv3/pytorch/bin/activate|source /torch/venv3/pytorch_infer/bin/activate|' /root/.bashrc

RUN apt-get update && \
    apt-get install -y ffmpeg libsox-dev redis && \
    rm -rf /var/lib/apt/lists/*

COPY wav /workspace/wav
COPY GPT-SoVITS /workspace/GPT-SoVITS

RUN pip install -r GPT-SoVITS/extra-req.txt --no-deps \
    && pip install -r GPT-SoVITS/requirements.txt

COPY launch_gsv.sh /workspace/
ENTRYPOINT ["/bin/bash", "launch_gsv.sh"]