mirror of
https://github.com/jamiepine/voicebox.git
synced 2026-09-15 04:40:40 -07:00
Jamie Pine
530ee407ee
- Added exclusions for 'torch.utils.tensorboard', 'scipy', 'PIL', 'tkinter', 'unittest', and 'test' to reduce bundle size and improve build efficiency.
94 lines
3.1 KiB
Python
94 lines
3.1 KiB
Python
"""
|
|
PyInstaller build script for creating standalone Python server binary.
|
|
"""
|
|
|
|
import PyInstaller.__main__
|
|
import os
|
|
from pathlib import Path
|
|
|
|
|
|
def build_server():
|
|
"""Build Python server as standalone binary."""
|
|
backend_dir = Path(__file__).parent
|
|
|
|
# Check for local editable qwen_tts install
|
|
local_qwen_path = Path.home() / 'Projects' / 'voice' / 'Qwen3-TTS'
|
|
|
|
# PyInstaller arguments
|
|
args = [
|
|
'server.py', # Use server.py as entry point instead of main.py
|
|
'--onefile',
|
|
'--name', 'voicebox-server',
|
|
]
|
|
|
|
# Add local qwen_tts path if it exists (for editable installs)
|
|
if local_qwen_path.exists():
|
|
args.extend(['--paths', str(local_qwen_path)])
|
|
print(f"Using local qwen_tts source from: {local_qwen_path}")
|
|
|
|
# Exclude unnecessary modules to reduce size
|
|
args.extend([
|
|
'--exclude-module', 'matplotlib',
|
|
'--exclude-module', 'IPython',
|
|
'--exclude-module', 'notebook',
|
|
'--exclude-module', 'pytest',
|
|
'--exclude-module', 'setuptools',
|
|
'--exclude-module', 'torch.distributions',
|
|
'--exclude-module', 'torch.testing',
|
|
'--exclude-module', 'tensorboard',
|
|
'--exclude-module', 'torch.utils.tensorboard',
|
|
'--exclude-module', 'scipy',
|
|
'--exclude-module', 'PIL',
|
|
'--exclude-module', 'tkinter',
|
|
'--exclude-module', 'unittest',
|
|
'--exclude-module', 'test',
|
|
])
|
|
|
|
# Add hidden imports
|
|
args.extend([
|
|
'--hidden-import', 'backend',
|
|
'--hidden-import', 'backend.main',
|
|
'--hidden-import', 'backend.config',
|
|
'--hidden-import', 'backend.database',
|
|
'--hidden-import', 'backend.models',
|
|
'--hidden-import', 'backend.profiles',
|
|
'--hidden-import', 'backend.history',
|
|
'--hidden-import', 'backend.tts',
|
|
'--hidden-import', 'backend.transcribe',
|
|
'--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.validation',
|
|
'--hidden-import', 'torch',
|
|
'--hidden-import', 'transformers',
|
|
'--hidden-import', 'fastapi',
|
|
'--hidden-import', 'uvicorn',
|
|
'--hidden-import', 'sqlalchemy',
|
|
'--hidden-import', 'librosa',
|
|
'--hidden-import', 'soundfile',
|
|
'--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',
|
|
'--noconfirm',
|
|
'--clean',
|
|
])
|
|
|
|
# Change to backend directory
|
|
os.chdir(backend_dir)
|
|
|
|
# Run PyInstaller
|
|
PyInstaller.__main__.run(args)
|
|
|
|
print(f"Binary built in {backend_dir / 'dist' / 'voicebox-server'}")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
build_server()
|