From c80c0e0fc0d33e5e680e4dd0e32a6f3a17267471 Mon Sep 17 00:00:00 2001
From: Douglas Yang <dyang@college.harvard.edu>
Date: Mon, 5 Jan 2026 20:10:13 -0800
Subject: [PATCH] fix: adding multi-threading for kimi weights loading in pr
 test (#16538)

---
 test/srt/models/test_kimi_k2_models.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/test/srt/models/test_kimi_k2_models.py b/test/srt/models/test_kimi_k2_models.py
index fc5e4c387..6a2fbed71 100644
--- a/test/srt/models/test_kimi_k2_models.py
+++ b/test/srt/models/test_kimi_k2_models.py
@@ -28,6 +28,8 @@ class TestKimiK2Thinking(CustomTestCase):
             "kimi_k2",
             "--reasoning-parser",
             "kimi_k2",
+            "--model-loader-extra-config",
+            '{"enable_multithread_load": true, "num_threads": 64}',
         ]
         cls.process = popen_launch_server(
             cls.model,