mirror of
https://github.com/jamiepine/voicebox.git
synced 2026-09-17 13:50:39 -07:00
Jamie Pine
61dabe7382
- Added support for packaging provider archives in the release workflow, creating platform-specific zip and tar.gz files for distribution. - Updated the `.gitignore` to exclude `.spec` files. - Introduced a new `CudaDownloadSection` component to manage CUDA downloads, including progress tracking and error handling. - Refactored provider download logic to handle archive extraction and cleanup after download. - Improved subprocess output handling in the provider manager for better logging and error reporting.
91 lines
2.9 KiB
Python
91 lines
2.9 KiB
Python
"""
|
|
PyInstaller build script for PyTorch CPU provider.
|
|
"""
|
|
|
|
import PyInstaller.__main__
|
|
import os
|
|
import platform
|
|
from pathlib import Path
|
|
|
|
|
|
def build_provider():
|
|
"""Build PyTorch CPU provider as standalone binary."""
|
|
provider_dir = Path(__file__).parent
|
|
backend_dir = provider_dir.parent.parent / "backend"
|
|
|
|
# PyInstaller arguments
|
|
args = [
|
|
'main.py',
|
|
'--onedir', # Changed from --onefile to work around Windows extraction issues
|
|
'--name', 'tts-provider-pytorch-cpu',
|
|
]
|
|
|
|
# Add backend to path
|
|
args.extend([
|
|
'--paths', str(backend_dir.parent),
|
|
])
|
|
|
|
# Add hidden imports
|
|
args.extend([
|
|
'--hidden-import', 'backend',
|
|
'--hidden-import', 'backend.backends',
|
|
'--hidden-import', 'backend.backends.pytorch_backend',
|
|
'--hidden-import', 'backend.config',
|
|
'--hidden-import', 'backend.utils.audio',
|
|
'--hidden-import', 'backend.utils.cache',
|
|
'--hidden-import', 'backend.utils.progress',
|
|
'--hidden-import', 'backend.utils.hf_progress',
|
|
'--hidden-import', 'backend.utils.tasks',
|
|
'--hidden-import', 'torch',
|
|
'--hidden-import', 'transformers',
|
|
'--hidden-import', 'qwen_tts',
|
|
'--hidden-import', 'qwen_tts.inference',
|
|
'--hidden-import', 'qwen_tts.inference.qwen3_tts_model',
|
|
'--hidden-import', 'qwen_tts.inference.qwen3_tts_tokenizer',
|
|
'--hidden-import', 'qwen_tts.core',
|
|
'--hidden-import', 'qwen_tts.cli',
|
|
'--copy-metadata', 'qwen-tts',
|
|
'--collect-submodules', 'qwen_tts',
|
|
'--collect-data', 'qwen_tts',
|
|
'--hidden-import', 'pkg_resources.extern',
|
|
'--collect-submodules', 'jaraco',
|
|
'--hidden-import', 'fastapi',
|
|
'--hidden-import', 'uvicorn',
|
|
# Critical uvicorn imports for PyInstaller
|
|
'--hidden-import', 'uvicorn.logging',
|
|
'--hidden-import', 'uvicorn.loops',
|
|
'--hidden-import', 'uvicorn.loops.auto',
|
|
'--hidden-import', 'uvicorn.protocols',
|
|
'--hidden-import', 'uvicorn.protocols.http',
|
|
'--hidden-import', 'uvicorn.protocols.http.auto',
|
|
'--hidden-import', 'uvicorn.protocols.websockets',
|
|
'--hidden-import', 'uvicorn.protocols.websockets.auto',
|
|
'--hidden-import', 'uvicorn.lifespan',
|
|
'--hidden-import', 'uvicorn.lifespan.on',
|
|
'--collect-submodules', 'uvicorn',
|
|
'--hidden-import', 'soundfile',
|
|
'--hidden-import', 'numpy',
|
|
'--hidden-import', 'librosa',
|
|
])
|
|
|
|
args.extend([
|
|
'--noconfirm',
|
|
'--clean',
|
|
])
|
|
|
|
# Change to provider directory
|
|
os.chdir(provider_dir)
|
|
|
|
# Run PyInstaller
|
|
PyInstaller.__main__.run(args)
|
|
|
|
binary_name = 'tts-provider-pytorch-cpu'
|
|
if platform.system() == "Windows":
|
|
binary_name += '.exe'
|
|
|
|
print(f"Binary built in {provider_dir / 'dist' / binary_name}")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
build_provider()
|