neonbjb/tortoise-tts
1
0
Fork 0
mirror of https://github.com/neonbjb/tortoise-tts.git synced 2026-03-18 19:24:40 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: docker flow

Browse source
This commit is contained in:
nikitalokhmachev-ai 2024-07-29 22:41:46 -04:00
parent 86100b6a23
commit ef5e8e2c52
2 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
Dockerfile
View file

@ -26,9 +26,9 @@ RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -
# --login option used to source bashrc (thus activating conda env) at every RUN statement
SHELL ["/bin/bash", "--login", "-c"]
RUN conda create --name tortoise python=3.9 numba inflect \
RUN conda create --name tortoise python=3.9 numba inflect -y \
&& conda activate tortoise \
&& conda install pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=12.1 -c pytorch -c nvidia \
&& conda install transformers=4.31.0 \
&& conda install --yes pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=12.1 -c pytorch -c nvidia \
&& conda install --yes transformers=4.31.0 \
&& cd /app \
&& python setup.py install

3
setup.py
View file

@ -28,7 +28,8 @@ setuptools.setup(
'scipy',
'librosa',
'transformers==4.31.0',
'tokenizers',
'tokenizers==0.13.4.rc3',
'scipy==1.13.1'
# 'deepspeed==0.8.3',
],
classifiers=[