FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-py3.10-cuda12.1-ubuntu22.04-devel as base
ARG IMAGE=voicechat2
ARG IMAGE_UPPER=voicechat2
ARG BRANCH=gpu
RUN cd /root && git clone -b $BRANCH http://developer.hpccube.com/codes/chenpangpang/$IMAGE.git
WORKDIR /root/$IMAGE/$IMAGE_UPPER

#########
# Prod  #
#########
FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-py3.10-cuda12.1-ubuntu22.04-devel
ARG IMAGE=voicechat2
ARG IMAGE_UPPER=voicechat2
COPY chenyh/$IMAGE/Meta-Llama-3-8B-Instruct-Q4_K_M.gguf /root/$IMAGE_UPPER/llama.cpp/Meta-Llama-3-8B-Instruct-Q4_K_M.gguf
COPY chenyh/$IMAGE/tts_models--en--vctk--vits /root/.local/share/tts/tts_models--en--vctk--vits
COPY chenyh/$IMAGE/distil-whisper/large-v2 /root/$IMAGE_UPPER/distil-whisper/large-v2
RUN apt-get update && apt-get install -y build-essential  byobu curl wget espeak-ng ffmpeg libopus0 libopus-dev

COPY --from=base /root/$IMAGE/$IMAGE_UPPER /root/$IMAGE_UPPER
WORKDIR /root/$IMAGE_UPPER
# 如果直接复制库，则缺少可执行程序，故在子镜像中安装
RUN pip install -r requirements.txt
COPY --from=base /root/$IMAGE/启动器.ipynb /root/$IMAGE/start.sh /root/
COPY --from=base /root/$IMAGE/assets /root/assets
# 需要在同种型号gpu上编译
# RUN cd llama.cpp && make GGML_CUDA=1 -j