Spaces:
Running
Running
File size: 2,432 Bytes
1d8403e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 | # ==================================================================================================
# DEEPFAKE AUDIO - utils/default_models.py (Pre-trained Asset Management)
# ==================================================================================================
#
# 📝 DESCRIPTION
# This module manages the lifecycle of the default pre-trained neural weights for the
# encoder, synthesizer, and vocoder. it ensures that the necessary model assets are
# synchronized from the Hugging Face hub and verified for integrity before execution.
#
# 👤 AUTHORS
# - Amey Thakur (https://github.com/Amey-Thakur)
# - Mega Satish (https://github.com/msatmod)
#
# 🤝🏻 CREDITS
# Original Real-Time Voice Cloning methodology by CorentinJ
# Repository: https://github.com/CorentinJ/Real-Time-Voice-Cloning
#
# 🔗 PROJECT LINKS
# Repository: https://github.com/Amey-Thakur/DEEPFAKE-AUDIO
# Video Demo: https://youtu.be/i3wnBcbHDbs
# Research: https://github.com/Amey-Thakur/DEEPFAKE-AUDIO/blob/main/DEEPFAKE-AUDIO.ipynb
#
# 📜 LICENSE
# Released under the MIT License
# Release Date: 2021-02-06
# ==================================================================================================
from pathlib import Path
from huggingface_hub import hf_hub_download
HUGGINGFACE_REPO = "CorentinJ/SV2TTS"
default_models = {
"encoder": 17090379,
"synthesizer": 370554559,
"vocoder": 53845290,
}
def _download_model(model_name: str, target_dir: Path):
hf_hub_download(
repo_id=HUGGINGFACE_REPO,
revision="main",
filename=f"{model_name}.pt",
local_dir=str(target_dir),
local_dir_use_symlinks=False,
)
def ensure_default_models(models_dir: Path):
target_dir = models_dir / "default"
target_dir.mkdir(parents=True, exist_ok=True)
for model_name, expected_size in default_models.items():
target_path = target_dir / f"{model_name}.pt"
if target_path.exists():
if target_path.stat().st_size == expected_size:
continue
print(f"File {target_path} is not of expected size, redownloading...")
_download_model(model_name, target_dir)
assert target_path.exists() and target_path.stat().st_size == expected_size, (
f"Download for {target_path.name} failed. You may download models manually instead.\n"
f"https://huggingface.co/{HUGGINGFACE_REPO}"
)
|