Add Russian GPT-SoVITS weights

Dockerfile

@@ -26,18 +26,42 @@ WORKDIR /app
 RUN git clone --depth 1 https://github.com/RVC-Boss/GPT-SoVITS.git .
 
 # Hugging Face Docker Spaces route traffic to port 7860.
+# The Russian model card requires vocab size 753 and Russian base checkpoints.
 RUN python - <<'PY'
 from pathlib import Path
-path = Path('config.py')
-text = path.read_text()
-text = text.replace('webui_port_main = 9874', 'webui_port_main = 7860')
-path.write_text(text)
+
+replacements = {
+    'webui_port_main = 9874': 'webui_port_main = 7860',
+    's1v3.ckpt': 's1v3-ru.ckpt',
+    's2Gv2ProPlus.pth': 's2Gv2ProPlus-rus.pth',
+    'phoneme_vocab_size: 732': 'phoneme_vocab_size: 753',
+    'phoneme_vocab_size = 732': 'phoneme_vocab_size = 753',
+}
+for path in [Path('config.py'), *Path('GPT_SoVITS/configs').glob('*.yaml')]:
+    if not path.exists():
+        continue
+    text = path.read_text()
+    for old, new in replacements.items():
+        text = text.replace(old, new)
+    path.write_text(text)
 PY
 
 RUN conda create -y -n GPTSoVits python=3.10 && conda clean -afy
 RUN source /opt/conda/etc/profile.d/conda.sh && \
     conda activate GPTSoVits && \
-    bash install.sh --device CPU --source HF
+    bash install.sh --device CPU --source HF && \
+    pip install --no-cache-dir epitran
+
+# Russian GPT-SoVITS weights from https://huggingface.co/fuckSelf/GPT-SoVITS-Russian
+RUN mkdir -p GPT_SoVITS/pretrained_models/v2Pro GPT_weights_v2ProPlus SoVITS_weights_v2ProPlus && \
+    curl -L --retry 5 -o GPT_SoVITS/pretrained_models/s1v3-ru.ckpt \
+      https://huggingface.co/fuckSelf/GPT-SoVITS-Russian/resolve/main/s1v3-ru.ckpt && \
+    curl -L --retry 5 -o GPT_SoVITS/pretrained_models/v2Pro/s2Gv2ProPlus-rus.pth \
+      https://huggingface.co/fuckSelf/GPT-SoVITS-Russian/resolve/main/s2Gv2ProPlus-rus.pth && \
+    curl -L --retry 5 -o GPT_weights_v2ProPlus/ru-base-e20.ckpt \
+      https://huggingface.co/fuckSelf/GPT-SoVITS-Russian/resolve/main/ru-base-e20.ckpt && \
+    curl -L --retry 5 -o SoVITS_weights_v2ProPlus/ru-base_e10_s123570.pth \
+      https://huggingface.co/fuckSelf/GPT-SoVITS-Russian/resolve/main/ru-base_e10_s123570.pth
 
 EXPOSE 7860
 CMD ["bash", "-lc", "source /opt/conda/etc/profile.d/conda.sh && conda activate GPTSoVits && python webui.py"]