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

Merge pull request #809 from nikitalokhmachev-ai/docker-fix

Browse source 
Fixing issues with Docker Setup
This commit is contained in:
manmay nakhashi 2024-08-13 15:17:57 +05:30 committed by GitHub
parent 86100b6a23 ef5e8e2c52
commit 2d6d3868f9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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=[