mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-09 00:09:55 +00:00
Brandon Thomas
30 lines
1.4 KiB
Docker
30 lines
1.4 KiB
Docker
FROM nvidia/cuda:11.0.3-base-ubuntu20.04
|
|
ENV TZ=Etc/GMT
|
|
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
|
|
RUN apt-get update && apt-get install -y git ffmpeg python3.8 python3-pip curl
|
|
RUN git clone -q https://github.com/SortAnon/ControllableTalkNet /talknet
|
|
RUN git clone -q https://github.com/SortAnon/hifi-gan /talknet/hifi-gan
|
|
RUN python3.8 -m pip --no-cache-dir install -r "/talknet/requirements.txt" -f https://download.pytorch.org/whl/torch_stable.html
|
|
RUN python3.8 -m pip --no-cache-dir install git+https://github.com/SortAnon/NeMo.git
|
|
RUN python3.8 -m pip uninstall -y pesq
|
|
RUN python3.8 -m pip install pesq==0.0.2
|
|
RUN touch /talknet/is_docker
|
|
WORKDIR /talknet
|
|
EXPOSE 8050
|
|
|
|
RUN printf "#!/bin/bash \
|
|
\necho Updating TalkNet... \
|
|
\ngit -C /talknet reset --hard origin/main -q \
|
|
\ngit -C /talknet pull origin main -q \
|
|
\necho Updating HiFi-GAN... \
|
|
\ngit -C /talknet reset --hard origin/main -q \
|
|
\ngit -C /talknet pull origin main -q \
|
|
\necho Updating Python dependencies... \
|
|
\npython3.8 -m pip --quiet --no-cache-dir install -r /talknet/requirements.txt -f https://download.pytorch.org/whl/torch_stable.html \
|
|
\npython3.8 -m pip --quiet --no-cache-dir install git+https://github.com/SortAnon/NeMo.git \
|
|
\necho Launching TalkNet... \
|
|
\npython3.8 talknet_offline.py\n" > /talknet/docker_launch.sh
|
|
|
|
RUN chmod +x /talknet/docker_launch.sh
|
|
CMD ["./docker_launch.sh"]
|