Labyricorn/voicebox
1
0
Fork 0
mirror of https://github.com/jamiepine/voicebox.git synced 2026-09-16 21:30:39 -07:00
Code Issues Packages Projects Releases Wiki Activity

Exclude CUDA libraries and torch compiler modules to reduce bundle size

Browse Source
This commit is contained in:
Jamie Pine
2026-01-25 13:13:24 -08:00
parent a57e7dbc54
commit b479178e91
1 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/build_binary.py
+16
View File
@@ -45,6 +45,22 @@ def build_server():
'--exclude-module', 'sklearn',
'--exclude-module', 'pandas',
'--exclude-module', 'torchaudio',
'--exclude-module', 'triton',
'--exclude-module', 'torch.distributed',
'--exclude-module', 'torch.jit',
'--exclude-module', 'torch._dynamo',
'--exclude-module', 'torch._inductor',
'--exclude-module', 'nvidia.cuda_nvrtc',
'--exclude-module', 'nvidia.cuda_runtime',
'--exclude-module', 'nvidia.cuda_cupti',
'--exclude-module', 'nvidia.cudnn',
'--exclude-module', 'nvidia.cublas',
'--exclude-module', 'nvidia.cufft',
'--exclude-module', 'nvidia.curand',
'--exclude-module', 'nvidia.cusolver',
'--exclude-module', 'nvidia.cusparse',
'--exclude-module', 'nvidia.nccl',
'--exclude-module', 'nvidia.nvtx',
])
# Add hidden imports