[Spec][DSA] Add --speculative-dsa-topk-backend (#36313)

2026-08-25 23:35:03 -07:00
parent 0c84eaba7f
commit 3c9febc68b
9 changed files with 51 additions and 11 deletions
@@ -180,6 +180,7 @@ class MockModelRunner:
        self.config = {**DEFAULT_CONFIG, **(config or {})}
        self.dtype = self.config["dtype"]
        self.kv_cache_dtype = self.config["kv_cache_dtype"]
+        self.is_draft_worker = False
        self.is_hybrid_swa = False

        # Model configuration