Feat: slice-and-transcribe

initialize.py

@@ -36,6 +36,34 @@ def download_pretrained_models():
             )
 
 
+def download_jvnv_models():
+    files = [
+        "jvnv-F1/config.json",
+        "jvnv-F1/jvnv-F1.safetensors",
+        "jvnv-F1/style_vectors.npy",
+        "jvnv-F2/config.json",
+        "jvnv-F2/jvnv-F2.safetensors",
+        "jvnv-F2/style_vectors.npy",
+        "jvnv-M1/config.json",
+        "jvnv-M1/jvnv-M1.safetensors",
+        "jvnv-M1/style_vectors.npy",
+        # "jvnv-M2/config.json",
+        # "jvnv-M2/jvnv-M2.safetensors",
+        # "jvnv-M2/style_vectors.npy",
+    ]
+    for file in files:
+        if not Path(f"model_assets/{file}").exists():
+            logger.info(f"Downloading {file}")
+            hf_hub_download(
+                "litagin/style_bert_vits2_jvnv",
+                file,
+                local_dir="model_assets",
+                local_dir_use_symlinks=False,
+            )
+
+
 download_bert_models()
 
 download_pretrained_models()
+
+download_jvnv_models()