[Spec] Fold the DFLASH worker base into DFlashWorkerV2 on BaseSpecWorker (#27950)

2026-06-11 14:54:28 -07:00
parent acdb39edd2
commit fee717f303
3 changed files with 999 additions and 1022 deletions
@@ -1172,9 +1172,9 @@ class Scheduler(
        self.device_module = torch.get_device_module(self.device)

        # FutureMap is always-on: input_ids relay used in both modes.
-        # Workers not on BaseSpecWorker (e.g. NGRAM / DFLASH) lack the
-        # override; fall back to target-only so the helper still produces a
-        # safe decision (no accidental opt-out for unaudited shapes).
+        # Workers without the spec_v2_attn_backends override fall back to
+        # target-only so the helper still produces a safe decision (no
+        # accidental opt-out for unaudited shapes).
        if self.draft_worker is not None:
            attn_backends = getattr(
                self.draft_worker,