Spaces:
Running
Running
File size: 3,547 Bytes
75c5859 3f9cba0 |
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 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 |
import os
import shutil
import subprocess
import argparse
import sys
def install_editable_package():
"""Install the f5-tts package in editable mode without dependencies."""
try:
subprocess.run(
[sys.executable, "-m", "pip", "install", "-e", ".", "--no-deps"],
check=True,
capture_output=True,
text=True
)
print("Successfully installed f5-tts package in editable mode")
except subprocess.CalledProcessError as e:
print(f"Failed to install f5-tts package: {e.stderr}")
sys.exit(1)
def clone_repository(repo_url, target_dir, branch="main"):
"""Clone a git repository to the specified directory with specific branch."""
if os.path.exists(target_dir):
print(f"Directory {target_dir} already exists. Skipping clone.")
return
os.makedirs(os.path.dirname(target_dir), exist_ok=True)
try:
subprocess.run(
["git", "clone", "-b", branch, repo_url, target_dir],
check=True,
capture_output=True,
text=True
)
print(f"Successfully cloned {repo_url} (branch: {branch}) to {target_dir}")
except subprocess.CalledProcessError as e:
print(f"Failed to clone repository: {e.stderr}")
sys.exit(1)
def handle_remove_readonly(func, path, _):
os.chmod(path, 0o666)
func(path)
def setup_python_path():
"""Add the src directory to sys.path to allow module imports."""
src_dir = os.path.abspath("src")
if src_dir not in sys.path:
sys.path.insert(0, src_dir)
print(f"Added {src_dir} to sys.path")
else:
print(f"{src_dir} already in sys.path")
def main(args):
# Define target directories
temp_f5_tts_target_dir = os.path.join("src", "danhtran2mind_f5_tts")
bigvgan_target_dir = os.path.join("src", "third_party", "BigVGAN")
f5_tts_target_dir = os.path.join("src", "f5_tts")
# Clone F5-TTS repository
clone_repository(args.f5_tts_url, temp_f5_tts_target_dir, args.f5_tts_branch)
# Clone BigVGAN repository
clone_repository(args.bigvgan_url, bigvgan_target_dir, args.bigvgan_branch)
# Move the directory
shutil.move(os.path.join(temp_f5_tts_target_dir, "src", "f5_tts"), f5_tts_target_dir)
shutil.copytree(os.path.join(temp_f5_tts_target_dir, "data"), "./data", dirs_exist_ok=True)
# Remove the nested f5_tts directory
if os.path.exists(os.path.join(f5_tts_target_dir, "f5_tts")):
shutil.rmtree(os.path.join(f5_tts_target_dir, "f5_tts"), onerror=handle_remove_readonly)
# Remove the parent directory
shutil.rmtree(temp_f5_tts_target_dir, onerror=handle_remove_readonly)
# Set up Python path to include src directory
install_editable_package()
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Clone F5-TTS and BigVGAN repositories and set up Python path")
parser.add_argument(
"--f5-tts-url",
default="https://github.com/danhtran2mind/F5-TTS",
help="URL for F5-TTS repository"
)
parser.add_argument(
"--bigvgan-url",
default="https://github.com/NVIDIA/BigVGAN",
help="URL for BigVGAN repository"
)
parser.add_argument(
"--f5-tts-branch",
default="main",
help="Branch for F5-TTS repository"
)
parser.add_argument(
"--bigvgan-branch",
default="main",
help="Branch or commit for BigVGAN repository"
)
args = parser.parse_args()
main(args) |