From b479178e919f123eef0522b6b727b4b04817adf3 Mon Sep 17 00:00:00 2001 From: Jamie Pine Date: Sun, 25 Jan 2026 13:13:24 -0800 Subject: [PATCH] Exclude CUDA libraries and torch compiler modules to reduce bundle size --- backend/build_binary.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/backend/build_binary.py b/backend/build_binary.py index 4793ca5f..31e7ba8d 100644 --- a/backend/build_binary.py +++ b/backend/build_binary.py @@ -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