From 631f72da394b61f3328420d49add3ce733322255 Mon Sep 17 00:00:00 2001
From: litagin02 <litagin02@gmail.com>
Date: Fri, 2 Feb 2024 22:21:49 +0900
Subject: [PATCH] Add downloading pretrained jp extra

---
 initialize.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/initialize.py b/initialize.py
index 1656c26..b236019 100644
--- a/initialize.py
+++ b/initialize.py
@@ -36,6 +36,20 @@ def download_pretrained_models():
             )
 
 
+def download_jp_extra_pretrained_models():
+    files = ["G_0.safetensors", "D_0.safetensors", "WD_0.safetensors"]
+    local_path = Path("pretrained_jp_extra")
+    for file in files:
+        if not Path(local_path).joinpath(file).exists():
+            logger.info(f"Downloading pretrained {file}")
+            hf_hub_download(
+                "litagin/Style-Bert-VITS2-2.0-base-JP-Extra",
+                file,
+                local_dir=local_path,
+                local_dir_use_symlinks=False,
+            )
+
+
 def download_jvnv_models():
     files = [
         "jvnv-F1/config.json",
@@ -66,4 +80,6 @@ download_bert_models()
 
 download_pretrained_models()
 
+download_jp_extra_pretrained_models()
+
 download_jvnv_models()