DeanoCalver commited on
Commit
31c489c
·
verified ·
1 Parent(s): 38bb126

Increase default live decode budget

Browse files
demo_controller.py CHANGED
@@ -11,7 +11,7 @@ from engines.compare import build_summary_sentence, compute_agreement, compute_m
11
  from engines.dense_runner import run_dense_live
12
  from engines.dotcache_runner import run_dotcache_live
13
  from engines.fixture_builder import FIXTURE_VERSION, build_fixture_result
14
- from engines.live_request import resolve_live_runtime_settings
15
  from engines.presets import (
16
  BIT_WIDTHS,
17
  CONTEXT_LENGTHS,
@@ -250,7 +250,7 @@ def _execute_live_request(request: DemoRequest, cache: ResultCache) -> Controlle
250
  def _validate_live_request(request_payload: dict[str, Any]) -> None:
251
  resolve_live_runtime_settings(
252
  request_payload,
253
- decode_steps=int(os.getenv("DOTCACHE_SPACE_LIVE_DECODE_STEPS", "24")),
254
  max_live_context=int(os.getenv("DOTCACHE_SPACE_MAX_LIVE_CONTEXT", "4096")),
255
  )
256
 
 
11
  from engines.dense_runner import run_dense_live
12
  from engines.dotcache_runner import run_dotcache_live
13
  from engines.fixture_builder import FIXTURE_VERSION, build_fixture_result
14
+ from engines.live_request import DEFAULT_LIVE_DECODE_STEPS, resolve_live_runtime_settings
15
  from engines.presets import (
16
  BIT_WIDTHS,
17
  CONTEXT_LENGTHS,
 
250
  def _validate_live_request(request_payload: dict[str, Any]) -> None:
251
  resolve_live_runtime_settings(
252
  request_payload,
253
+ decode_steps=int(os.getenv("DOTCACHE_SPACE_LIVE_DECODE_STEPS", str(DEFAULT_LIVE_DECODE_STEPS))),
254
  max_live_context=int(os.getenv("DOTCACHE_SPACE_MAX_LIVE_CONTEXT", "4096")),
255
  )
256
 
engines/live_request.py CHANGED
@@ -14,6 +14,8 @@ SHORTLIST_TOP_K = {
14
  "aggressive": 2,
15
  }
16
 
 
 
17
  SELECTIVE_EXACT_K_LAYERS = (19, 23)
18
 
19
 
 
14
  "aggressive": 2,
15
  }
16
 
17
+ DEFAULT_LIVE_DECODE_STEPS = 48
18
+
19
  SELECTIVE_EXACT_K_LAYERS = (19, 23)
20
 
21
 
scripts/space_dense_runner.py CHANGED
@@ -10,7 +10,7 @@ if str(REPO_ROOT) not in sys.path:
10
  sys.path.insert(0, str(REPO_ROOT))
11
 
12
  from dotcache.integrations.qwen35 import Qwen35TextHarness # noqa: E402
13
- from engines.live_request import resolve_live_runtime_settings # noqa: E402
14
  from scripts.space_runner_common import ( # noqa: E402
15
  configure_model_cache_env,
16
  decode_generated_text,
@@ -51,7 +51,7 @@ def main() -> int:
51
  request = load_request_from_stdin()
52
  settings = resolve_live_runtime_settings(
53
  request,
54
- decode_steps=int(os.getenv("DOTCACHE_SPACE_LIVE_DECODE_STEPS", "24")),
55
  max_live_context=int(os.getenv("DOTCACHE_SPACE_MAX_LIVE_CONTEXT", "4096")),
56
  )
57
 
 
10
  sys.path.insert(0, str(REPO_ROOT))
11
 
12
  from dotcache.integrations.qwen35 import Qwen35TextHarness # noqa: E402
13
+ from engines.live_request import DEFAULT_LIVE_DECODE_STEPS, resolve_live_runtime_settings # noqa: E402
14
  from scripts.space_runner_common import ( # noqa: E402
15
  configure_model_cache_env,
16
  decode_generated_text,
 
51
  request = load_request_from_stdin()
52
  settings = resolve_live_runtime_settings(
53
  request,
54
+ decode_steps=int(os.getenv("DOTCACHE_SPACE_LIVE_DECODE_STEPS", str(DEFAULT_LIVE_DECODE_STEPS))),
55
  max_live_context=int(os.getenv("DOTCACHE_SPACE_MAX_LIVE_CONTEXT", "4096")),
56
  )
57
 
scripts/space_dotcache_runner.py CHANGED
@@ -11,7 +11,11 @@ if str(REPO_ROOT) not in sys.path:
11
 
12
  from benchmarks.bench_qwen35_attention_subset_dotcache_serving import _build_dotcache_config # noqa: E402
13
  from dotcache.integrations.qwen35 import Qwen35AttentionSubsetDotCacheHarness # noqa: E402
14
- from engines.live_request import resolve_live_runtime_settings, selective_exact_k_overrides # noqa: E402
 
 
 
 
15
  from scripts.space_runner_common import ( # noqa: E402
16
  configure_model_cache_env,
17
  decode_generated_text,
@@ -149,7 +153,7 @@ def main() -> int:
149
  request = load_request_from_stdin()
150
  settings = resolve_live_runtime_settings(
151
  request,
152
- decode_steps=int(os.getenv("DOTCACHE_SPACE_LIVE_DECODE_STEPS", "24")),
153
  max_live_context=int(os.getenv("DOTCACHE_SPACE_MAX_LIVE_CONTEXT", "4096")),
154
  )
155
 
 
11
 
12
  from benchmarks.bench_qwen35_attention_subset_dotcache_serving import _build_dotcache_config # noqa: E402
13
  from dotcache.integrations.qwen35 import Qwen35AttentionSubsetDotCacheHarness # noqa: E402
14
+ from engines.live_request import ( # noqa: E402
15
+ DEFAULT_LIVE_DECODE_STEPS,
16
+ resolve_live_runtime_settings,
17
+ selective_exact_k_overrides,
18
+ )
19
  from scripts.space_runner_common import ( # noqa: E402
20
  configure_model_cache_env,
21
  decode_generated_text,
 
153
  request = load_request_from_stdin()
154
  settings = resolve_live_runtime_settings(
155
  request,
156
+ decode_steps=int(os.getenv("DOTCACHE_SPACE_LIVE_DECODE_STEPS", str(DEFAULT_LIVE_DECODE_STEPS))),
157
  max_live_context=int(os.getenv("DOTCACHE_SPACE_MAX_LIVE_CONTEXT", "4096")),
158
  )
159