From 72a4494c558db98eb5b8bfbf00b51c6b8f3102c7 Mon Sep 17 00:00:00 2001
From: lwroo <95639516+lwroo@users.noreply.github.com>
Date: Tue, 15 Aug 2023 04:38:42 +0200
Subject: [PATCH] chore: extend Dockerfile, add docker-compose.yml (#1009)

Co-authored-by: lwroo.local <me@lwroo.party>
---
 Dockerfile         | 16 ++++++++++++++++
 docker-compose.yml | 13 +++++++++++++
 2 files changed, 29 insertions(+)
 create mode 100644 docker-compose.yml

diff --git a/Dockerfile b/Dockerfile
index 49f62d5..65ffbc1 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -8,6 +8,22 @@ WORKDIR /app
 
 COPY . .
 
+RUN apt update && apt install -y -qq ffmpeg aria2
+
 RUN pip3 install -r requirements.txt
 
+RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main/pretrained_v2/D40k.pth -d pretrained_v2/ -o D40k.pth
+RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main/pretrained_v2/G40k.pth -d pretrained_v2/ -o G40k.pth
+RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main/pretrained_v2/f0D40k.pth -d pretrained_v2/ -o f0D40k.pth
+RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main/pretrained_v2/f0G40k.pth -d pretrained_v2/ -o f0G40k.pth
+
+RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main/uvr5_weights/HP2-人声vocals+非人声instrumentals.pth -d uvr5_weights/ -o HP2-人声vocals+非人声instrumentals.pth
+RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main/uvr5_weights/HP5-主旋律人声vocals+其他instrumentals.pth -d uvr5_weights/ -o HP5-主旋律人声vocals+其他instrumentals.pth
+
+RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main/hubert_base.pt -o hubert_base.pt
+
+RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main/rmvpe.pt -o rmvpe.pt
+
+VOLUME [ "/app/weights", "/app/opt" ]
+
 CMD ["python3", "infer-web.py"]
\ No newline at end of file
diff --git a/docker-compose.yml b/docker-compose.yml
new file mode 100644
index 0000000..4b40ec5
--- /dev/null
+++ b/docker-compose.yml
@@ -0,0 +1,13 @@
+version: "3.8"
+services:
+  rvc:
+    build:
+      context: .
+      dockerfile: Dockerfile
+    container_name: rvc
+    volumes:
+      - ./weights:/app/weights
+      - ./opt:/app/opt
+      # - ./dataset:/app/dataset # you can use this folder in order to provide your dataset for model training
+    ports:
+      - 7865:7865
\ No newline at end of file