Spaces:
Running
Running
| 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) |