diff --git a/test/registered/unit/test_model_overrides.py b/test/registered/unit/test_model_overrides.py index 81a820c74..b9e7c8e89 100644 --- a/test/registered/unit/test_model_overrides.py +++ b/test/registered/unit/test_model_overrides.py @@ -1737,6 +1737,7 @@ class TestGoldenModelOverrides(_IsolatedPublish): attention_backend=None, decode_attention_backend=None, prefill_attention_backend=None, + speculative_draft_attention_backend=None, page_size=1, ) defaults.update(kw)