storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-17 00:23:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

smaller tt2 images

Browse Source
This commit is contained in:
Kasisnu
2024-04-22 18:54:23 +05:30
parent d1a2de8c59
commit 951dbc94a5
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tts/tacotron2_v1_early_fakeyou/Dockerfile
+2 -2
View File
@@ -106,7 +106,7 @@ WORKDIR models/tts
# NB: Setuptools fix: https://github.com/tensorflow/tensorflow/issues/34302#issuecomment-554450289
RUN python3.6 -m pip install --upgrade pip setuptools \
&& python3.6 -m pip install -r requirements-tacotron-python36.txt \
&& python3.6 -m pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
&& python3.6 -m pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 && python3.6 -m pip cache purge
# Final image
# FROM ubuntu:xenial
@@ -140,7 +140,7 @@ RUN apt-get install -y libespeak-ng1
RUN python3 --version
# TODO: Use venv to do this instead.
RUN pip3 install gdown
RUN pip3 install gdown && pip3 cache purge
# NB(echelon,2022-07-14): Run a newer version of pytorch which will support RTX3090.
# Fixes bug, "NVIDIA GeForce RTX 3080 with CUDA capability sm_86 is not compatible with the current PyTorch installation."