mrq/ai-voice-cloning
31
74
Fork 43
Code Issues 222 Pull Requests 5 Packages Projects Releases Wiki Activity

Compare commits

base: mrq:d60a7f10f22833309de4d1a8b6726eab005888be
Branches Tags
mrq:master
..
compare: mrq:d815b387b573ac86e10846e6375f2c1f7e5eb666
Branches Tags
mrq:master

1 Commits
d60a7f10f2 ... d815b387b5

Author SHA1 Message Date
psr
d815b387b5 dockerfile init 2023-04-07 21:17:45 +00:00
3 changed files with 22 additions and 23 deletions
Show Stats Expand all files Collapse all files

25
Dockerfile
View File

@ -19,36 +19,17 @@ RUN python3 -m pip install --upgrade pip
RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118 RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
FROM stage1 AS stage2 FROM stage1 AS stage2
USER root
RUN mkdir /home/user/ai-voice-cloning RUN mkdir /home/user/ai-voice-cloning
RUN mkdir /home/user/ai-voice-cloning/modules
RUN chown -R user:user /home/user/ai-voice-cloning
USER user
WORKDIR /home/user/ai-voice-cloning WORKDIR /home/user/ai-voice-cloning
COPY --chown=user:user modules modules
ADD .gitmodules .gitmodules
ADD .git .git
ADD modules modules
USER root
RUN chown -R user:user /home/user/ai-voice-cloning
USER user
RUN git config --global --add safe.directory /home/user/ai-voice-cloning
RUN git submodule init
RUN git submodule update --remote
# install TorToiSe requirements
RUN python3 -m pip install -r ./modules/tortoise-tts/requirements.txt RUN python3 -m pip install -r ./modules/tortoise-tts/requirements.txt
# install TorToiSe FROM stage2 AS stage3
RUN python3 -m pip install -e ./modules/tortoise-tts/ RUN python3 -m pip install -e ./modules/tortoise-tts/
# instal DLAS requirements, last, because whisperx will break a dependency here
RUN python3 -m pip install -r ./modules/dlas/requirements.txt RUN python3 -m pip install -r ./modules/dlas/requirements.txt
# install DLAS
RUN python3 -m pip install -e ./modules/dlas/ RUN python3 -m pip install -e ./modules/dlas/
ADD requirements.txt requirements.txt ADD requirements.txt requirements.txt
# install local requirements
RUN python3 -m pip install -r ./requirements.txt RUN python3 -m pip install -r ./requirements.txt
ADD --chown=user:user . /home/user/ai-voice-cloning
ADD . /home/user/ai-voice-cloning
CMD ["python", "./src/main.py", "--listen", "0.0.0.0:7680"] CMD ["python", "./src/main.py", "--listen", "0.0.0.0:7680"]

4
setup-docker.sh Executable file
View File

@ -0,0 +1,4 @@
#!/bin/bash
git submodule init
git submodule update --remote
docker build -t ai-voice-cloning .

14
start-docker.sh Executable file
View File

@ -0,0 +1,14 @@
#!/bin/bash
CMD="python3 ./src/main.py $@"
# CMD="bash"
CPATH="/home/user/ai-voice-cloning"
docker run --rm --gpus all \
--mount "type=bind,src=$PWD/models,dst=$CPATH/models" \
--mount "type=bind,src=$PWD/training,dst=$CPATH/training" \
--mount "type=bind,src=$PWD/voices,dst=$CPATH/voices" \
--mount "type=bind,src=$PWD/bin,dst=$CPATH/bin" \
--workdir $CPATH \
--user "$(id -u):$(id -g)" \
--net host \
-it ai-voice-cloning $CMD