From 240b9b71a7d94165c6ffc9ac86cdd2ef56e2f0e1 Mon Sep 17 00:00:00 2001
From: Jamie Pine <ijamespine@me.com>
Date: Sun, 25 Jan 2026 21:44:12 -0800
Subject: [PATCH] Remove torch.testing exclusion - needed by torch internals

---
 backend/build_binary.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/backend/build_binary.py b/backend/build_binary.py
index 5f43652a..c4868bed 100644
--- a/backend/build_binary.py
+++ b/backend/build_binary.py
@@ -27,14 +27,14 @@ def build_server():
         print(f"Using local qwen_tts source from: {local_qwen_path}")
 
     # Exclude unnecessary modules to reduce size
+    # Be careful with torch exclusions - torch internally imports many of its own submodules
     args.extend([
         '--exclude-module', 'matplotlib',
         '--exclude-module', 'IPython',
         '--exclude-module', 'notebook',
         '--exclude-module', 'pytest',
         # Don't exclude setuptools - pkg_resources needs it
-        '--exclude-module', 'torch.distributions',
-        '--exclude-module', 'torch.testing',
+        # Don't exclude torch.testing - torch.autograd.gradcheck needs it
         '--exclude-module', 'tensorboard',
         '--exclude-module', 'torch.utils.tensorboard',
         '--exclude-module', 'scipy',